Changed statement slightly