diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 1edf943b4f015e3881ab0420bf69f3f21ac06456..b258519b680834d770e1823cb58a3aa4d9589b37 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -280,7 +280,7 @@ class TreeObsForRailEnv(ObservationBuilder):
                         min_fractional_speed = current_fractional_speed
 
                     # Other direction agents
-                    # TODO: This does not work as expected yet
+                    # TODO: Test that this behavior is as expected
                     other_agent_opposite_direction += self.location_has_agent[position] - \
                                                       self.location_has_agent_direction[
                                                           (position, direction)]