diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index bf9679f9d7d467ec1962aac4466aaa1154e443b5..8fa40c8eb3fe74e3fab41cc16acc93cd134b391b 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -896,6 +896,17 @@ def realistic_rail_generator(nr_start_goal=1, seed=0, add_max_dead_end=4, two_tr
         agents_target = []
         agents_direction = []
 
+        for a in range(len(agents_positions)):
+            cell_transitions = grid_map.get_transitions(agents_positions[a][0], agents_positions[a][1],
+                                                        agents_directions[a])
+            if np.sum(cell_transitions) == 0:
+                agents_directions[a] += 2
+                agents_directions[a] = agents_directions[a] % 4
+                cell_transitions = grid_map.get_transitions(agents_positions[a][0], agents_positions[a][1],
+                                                            agents_directions[a])
+                if np.sum(cell_transitions):
+                    print(a, "bug")
+
         for a in range(min(len(agents_targets), num_agents)):
             t = np.random.choice(range(len(agents_targets)))
             d = agents_targets[t]
@@ -1047,7 +1058,7 @@ def sparse_rail_generator(num_cities=100, num_intersections=10, num_trainstation
                                     0,
                                     width - 1)
                 while (station_x, station_y) in train_stations or (station_x, station_y) == node_positions[
-                        trainstation_node] or rail_array[(station_x, station_y)] != 0:
+                    trainstation_node] or rail_array[(station_x, station_y)] != 0:
                     station_x = np.clip(
                         node_positions[trainstation_node][0] + np.random.randint(-node_radius, node_radius),
                         0,