diff --git a/flatland/core/env_observation_builder.py b/flatland/core/env_observation_builder.py
index 848b184f7c99b991ebda3a62aad607ca1b718d86..ac5bfd8b867fed728de79fc2f2270381fdde04af 100644
--- a/flatland/core/env_observation_builder.py
+++ b/flatland/core/env_observation_builder.py
@@ -242,7 +242,7 @@ class TreeObsForRailEnv(ObservationBuilder):
         # Start from the current orientation, and see which transitions are available;
         # organize them as [left, forward, right, back], relative to the current orientation
         for branch_direction in [(orientation + 4 + i) % 4 for i in range(-1, 3)]:
-            #TODO: check if cell is a curve, then keep branch direction forward instead of left or right
+            # TODO: check if cell is a curve, then keep branch direction forward instead of left or right
             if self.env.rail.get_transition((position[0], position[1], orientation), branch_direction):
                 new_cell = self._new_position(position, branch_direction)
 
diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index 30e474d615fafbc29792345b43f92a1f28bb6a80..01562a7f24d558bf142b7b1c09ef5d0691541efb 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -905,7 +905,7 @@ class RailEnv(Environment):
     
     def reset(self, regen_rail=True, replace_agents=True):
         if regen_rail or self.rail is None:
-            #TODO: Import not only rail information but also start and goal positions
+            # TODO: Import not only rail information but also start and goal positions
             self.rail = self.rail_generator(self.width, self.height, self.num_resets)
             self.fill_valid_positions()