diff --git a/examples/flatland_2_0_example.py b/examples/flatland_2_0_example.py
index 7916c4e5057c5746b7d30d5ba2c9bfaab2d4b997..3ca6ab9d342c0b3a2907252264274899f8e519f7 100644
--- a/examples/flatland_2_0_example.py
+++ b/examples/flatland_2_0_example.py
@@ -30,7 +30,7 @@ speed_ration_map = {1.: 0.25,  # Fast passenger train
 
 env = RailEnv(width=50,
               height=50,
-              rail_generator=sparse_rail_generator(num_cities=10,  # Number of cities in map (where train stations are)
+              rail_generator=sparse_rail_generator(num_cities=2,  # Number of cities in map (where train stations are)
                                                    seed=1,  # Random seed
                                                    grid_mode=False,
                                                    max_inter_city_rails=2,
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index ba12736f6d3ad241f6b323bf1ef7204b036e3668..a1205580d89df199fc23b0f9c038764b3984044c 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -753,7 +753,7 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
                         tmp_direction = (out_direction - 1) % 4
                     while neighb_idx is None:
                         neighb_idx = neighbours[tmp_direction]
-                        tmp_direction = (out_direction + 1) % 4
+                        tmp_direction = (tmp_direction + 1) % 4
                     min_connection_dist = np.inf
                     for dir in range(4):
                         current_points = connection_points[neighb_idx][dir]
@@ -946,7 +946,7 @@ def sparse_rail_generator(num_cities=5, grid_mode=False, max_inter_city_rails=4,
 
             if direction_set == 4:
                 return closest_neighb
-
+        print(closest_neighb)
         return closest_neighb
 
     def argsort(seq):