diff --git a/flatland/core/env.py b/flatland/core/env.py
index a07a8095575a18eb20eb57a779fb3585dd01cf75..527ed30b3b630656015c2ba048c5453d712ae4fc 100644
--- a/flatland/core/env.py
+++ b/flatland/core/env.py
@@ -52,10 +52,10 @@ class Environment:
 
     def step(self, action_dict):
         """
-        Performs an environment step with simultaneous execution of actions for
-        agents in action_dict.
+        Environment step.
 
-        Returns observations from agents in the environment.
+        Performs an environment step with simultaneous execution of actions for
+        agents in action_dict. Returns observations from agents in the environment.
         The returns are dicts mapping from agent_id strings to values.
 
         Parameters