diff --git a/flatland/envs/grid4_generators_utils.py b/flatland/envs/grid4_generators_utils.py
index bdf49f87d4c33d899e5e0c53b7ee50df190a8c7f..0055b243668a4f3cd562958a59f52ba830af1c86 100644
--- a/flatland/envs/grid4_generators_utils.py
+++ b/flatland/envs/grid4_generators_utils.py
@@ -125,7 +125,7 @@ def connect_from_nodes(rail_trans, rail_array, start, end):
             if new_trans == 0:
                 # end-point
                 # need to flip direction because of how end points are defined
-                new_trans = rail_trans.set_transition(new_trans, current_dir, new_dir, 1)
+                new_trans = 0
             else:
                 # into existing rail
                 new_trans = rail_trans.set_transition(new_trans, current_dir, new_dir, 1)
@@ -187,7 +187,7 @@ def connect_to_nodes(rail_trans, rail_array, start, end):
             new_trans_e = rail_array[end_pos]
             if new_trans_e == 0:
                 # end-point
-                new_trans_e = rail_trans.set_transition(new_trans_e, new_dir, new_dir, 1)
+                new_trans_e = 0
             else:
                 # into existing rail
                 new_trans_e = rail_trans.set_transition(new_trans_e, new_dir, new_dir, 1)
diff --git a/tests/test_flatland_env_sparse_rail_generator.py b/tests/test_flatland_env_sparse_rail_generator.py
index 77b4c4af435f753396e10ceaba7c0b4b3ca8b2b3..927440805389ff8f6d7a4ab3672d86aaccb7f722 100644
--- a/tests/test_flatland_env_sparse_rail_generator.py
+++ b/tests/test_flatland_env_sparse_rail_generator.py
@@ -1,3 +1,5 @@
+import time
+
 import numpy as np
 
 from flatland.envs.generators import sparse_rail_generator, realistic_rail_generator
@@ -31,8 +33,9 @@ def test_sparse_rail_generator():
                                                        num_neighb=2,  # Number of connections to other cities
                                                        seed=5,  # Random seed
                                                        ),
-                  number_of_agents=10,
+                  number_of_agents=1,
                   obs_builder_object=GlobalObsForRailEnv())
     # reset to initialize agents_static
     env_renderer = RenderTool(env, gl="PILSVG", )
     env_renderer.render_env(show=True, show_observations=True, show_predictions=False)
+    time.sleep(2)