diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index 29545eacf0fd51f6e5cda3e29c5791c793eec7e8..6049e6db3f0aa1680934314a95c2e1533a634003 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -775,12 +775,12 @@ def realistic_rail_generator(nr_start_goal=1, seed=0):
                     idx_target += 1
 
             add_max_dead_end = 20
-            for pos_y in np.random.choice(np.arange(width-7)+3,add_max_dead_end,False):
-                pos_x = off_set+1
+            for pos_y in np.random.choice(np.arange(width - 7) + 3, add_max_dead_end, False):
+                pos_x = off_set + 1
                 if pos_x < height - 1:
                     ok = True
                     for k in range(6):
-                        c = (pos_x,pos_y-k+1)
+                        c = (pos_x, pos_y - k + 1)
                         ok &= grid_map.grid[c[0]][c[1]] == 0
                     if ok:
                         if np.random.random() < 0.5:
@@ -788,17 +788,16 @@ def realistic_rail_generator(nr_start_goal=1, seed=0):
                             goal_track = (pos_x, pos_y - 2)
                             new_path = connect_rail(rail_trans, rail_array, start_track, goal_track)
                             if len(new_path) > 0:
-                                c = (pos_x-1, pos_y-1)
+                                c = (pos_x - 1, pos_y - 1)
                                 make_switch_e_w(width, height, grid_map, c)
                         else:
                             start_track = (pos_x, pos_y)
                             goal_track = (pos_x, pos_y - 2)
                             new_path = connect_rail(rail_trans, rail_array, start_track, goal_track)
                             if len(new_path) > 0:
-                                c = (pos_x-1, pos_y+1)
+                                c = (pos_x - 1, pos_y + 1)
                                 make_switch_w_e(width, height, grid_map, c)
 
-
         agents_position = []
         agents_target = []
         agents_direction = []