diff --git a/flatland/envs/predictions.py b/flatland/envs/predictions.py
index d471596bba9e6128ea14f78bd8b625d321c255fe..671b349a4794f565b40e0d085393af6b92a08989 100644
--- a/flatland/envs/predictions.py
+++ b/flatland/envs/predictions.py
@@ -147,6 +147,9 @@ class ShortestPathPredictorForRailEnv(PredictionBuilder):
                             if target_dist < min_dist:
                                 min_dist = target_dist
                                 new_direction = direction
+                    if new_direction == None:
+                        prediction[index] = [index, *agent.position, agent.direction, RailEnvActions.STOP_MOVING]
+                        continue
                     new_position = get_new_position(agent.position, new_direction)
                 else:
                     raise Exception("No transition possible {}".format(cell_transitions))