diff --git a/flatland/core/transitions.py b/flatland/core/transitions.py
index a2bb657f46125a9a88954c88a0a942f368e84385..d70f8da096ed949b5b0541a7406fae370b271035 100644
--- a/flatland/core/transitions.py
+++ b/flatland/core/transitions.py
@@ -167,9 +167,11 @@ class GridTransitions(Transitions):
                  transitions,
                  allow_diagonal_transitions=False
                  ):
-        self.number_of_cell_neighbors = 4
+
         if allow_diagonal_transitions:
             self.number_of_cell_neighbors = 8
+        else:
+            self.number_of_cell_neighbors = 4
 
         self.transitions = transitions