diff --git a/flatland/envs/generators.py b/flatland/envs/generators.py
index de34246d9c326d2ec6e0ae042f93665ae4bf3b7e..6dff5d953afe7e8d9c3768fc58934f6ed899ab82 100644
--- a/flatland/envs/generators.py
+++ b/flatland/envs/generators.py
@@ -779,7 +779,7 @@ def realistic_rail_generator(nr_start_goal=1, seed=0,max_add_dead_end = 7):
 
             # add dead-end
             if True:
-                n = int(np.random.choice(np.arange(max_add_dead_end), 1)[0])
+                n = max_add_dead_end#int(np.random.choice(np.arange(max_add_dead_end-2)+1, 1)[0])
                 for pos_y in np.random.choice(np.arange(width - 7) + 3, n):
                     pos_x = off_set
                     pos_x1 = max(0, min(pos_x + 1, height - 1))