diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 4476bef5ac6ed271a860a6e9071f5b803448d75a..dc5741b95a39442065328667ccd9c53cda4c408e 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -489,7 +489,7 @@ class TreeObsForRailEnv(ObservationBuilder):
             depth += 1
             pow4 *= 4
         child_size = (len(tree) - num_features_per_node) // 4
-        tree_data = tree[0:num_features_per_node - 1].tolist()
+        tree_data = tree[0:num_features_per_node - 3].tolist()
         distance_data = [tree[num_features_per_node - 3]]
         agent_data = tree[-2:].tolist()
         for children in range(4):