Diego Costa's back in the race...