diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index ea9c16e1a7f5fcf91fc424559640679668092e51..d59ca7dc8f15d7f550e59f426a34cdd3275d954a 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -329,7 +329,7 @@ class RailEnv(Environment):
 
             # The train is broken
             if agent.malfunction_data['malfunction'] > 0:
-                # Last step of malfunction --> Agent starts moving again
+                # Last step of malfunction --> Agent starts moving again after getting fixed
                 if agent.malfunction_data['malfunction'] < 2:
                     agent.malfunction_data['malfunction'] -= 1
                     self.agents[i_agent].moving = True