Diego Costa, baja con la selección por una rotura fibrilar