diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 23a95ad63a798a7cc58e7272297e359c5c0f1f46..7bc2f994f8ae05f86e569cdb64bae08f6dc877f2 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -328,8 +328,6 @@ class TreeObsForRailEnv(ObservationBuilder):
                     # Cummulate the number of agents on branch with other direction
                     other_agent_opposite_direction += 1
 
-            if self.predictor:
-                continue
             # Register possible conflict
 
             if position in self.location_has_target: