diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index d6e1c34f0bd2fb9b0694b16bdfa93b3d00e4ff2d..a16fb6018a6354665a44c1b44cafd6975bb4e680 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -666,7 +666,7 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
                                     0,
                                     width - 1)
                 tries = 0
-                while (station_x, station_y) in train_stations \
+                while (station_x, station_y) in train_stations[trainstation_node] \
                     or (station_x, station_y) == node_positions[trainstation_node] \
                     or rail_array[(station_x, station_y)] != 0:  # noqa: E125