diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 007f6939f74590effffb22f9dbafc30809b6d693..4de614db4389556a877f4083fcdded6a7e3debdf 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -578,11 +578,8 @@ class TreeObsForRailEnv(ObservationBuilder):
             pow4 *= 4
         child_size = (len(tree) - num_features_per_node) // 4
         tree_data = tree[:4].tolist()
-        # print("data",tree_data)
         distance_data = [tree[4]]
-        #print("distance",distance_data)
         agent_data = tree[5:num_features_per_node].tolist()
-        #print("agent_data",agent_data)
         for children in range(4):
             child_tree = tree[(num_features_per_node + children * child_size):
                               (num_features_per_node + (children + 1) * child_size)]
diff --git a/flatland/envs/predictions.py b/flatland/envs/predictions.py
index cff4e00e14d73981dd623e796554f46a150c8a80..b6fe86316b5acb050465c3c726e5a04665696b45 100644
--- a/flatland/envs/predictions.py
+++ b/flatland/envs/predictions.py
@@ -108,7 +108,6 @@ class ShortestPathPredictorForRailEnv(PredictionBuilder):
         prediction_dict = {}
         agent_idx = 0
         for agent in agents:
-            action_priorities = [RailEnvActions.MOVE_FORWARD, RailEnvActions.MOVE_LEFT, RailEnvActions.MOVE_RIGHT]
             _agent_initial_position = agent.position
             _agent_initial_direction = agent.direction
             prediction = np.zeros(shape=(self.max_depth + 1, 5))