diff --git a/flatland/utils/editor.py b/flatland/utils/editor.py
index 1438a3ebea1738cf1feedb5a5ff38bcd308dee0e..d0f36a0deefb502017df1704a515ff4fda436a92 100644
--- a/flatland/utils/editor.py
+++ b/flatland/utils/editor.py
@@ -12,8 +12,8 @@ import flatland.utils.rendertools as rt
 from flatland.core.grid.grid4_utils import mirror
 from flatland.envs.agent_utils import EnvAgent
 from flatland.envs.observations import TreeObsForRailEnv
-from flatland.envs.rail_env import RailEnv, random_rail_generator
-from flatland.envs.rail_generators import complex_rail_generator, empty_rail_generator
+from flatland.envs.rail_env import RailEnv
+from flatland.envs.rail_generators import complex_rail_generator, empty_rail_generator, random_rail_generator
 
 
 class EditorMVC(object):