diff --git a/examples/training_navigation.py b/examples/training_navigation.py
index a7b2fbd391d5991d0d5a01da6eba425df14abcd9..4cd62eea73cd6fc66a6cf7f3a66912ce3d984d7a 100644
--- a/examples/training_navigation.py
+++ b/examples/training_navigation.py
@@ -23,7 +23,14 @@ transition_probability = [5,  # empty cell - Case 0
 env = RailEnv(width=10,
               height=10,
               rail_generator=random_rail_generator(cell_type_relative_proportion=transition_probability),
-              number_of_agents=5)
+              number_of_agents=2)
+
+env = RailEnv(width=20,
+              height=20,
+              rail_generator=rail_from_list_of_saved_GridTransitionMap_generator(
+                  ['../flatland/baselines/test-editor.npy']),
+              number_of_agents=1)
+
 env_renderer = RenderTool(env, gl="QT")
 handle = env.get_agent_handles()