diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 51f4cff78462f56642c493263eef872647d8f0bc..96d91579cf3c1c137291a95fb7bcb57d156457ee 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -585,12 +585,12 @@ class LocalObsForRailEnv(ObservationBuilder):
         # top_offset = max(0, agent.position[0] - 1 - self.view_radius)
         # bottom_offset = min(0, agent.position[0] + 1 + self.view_radius)
 
-        local_rail_obs = self.rail_obs[agent.position[0]: agent.position[0]+2*self.view_radius +1,
-                         agent.position[1]:agent.position[1]+2*self.view_radius +1]
+        local_rail_obs = self.rail_obs[agent.position[0]: agent.position[0]+2*self.view_radius + 1,
+                                       agent.position[1]:agent.position[1]+2*self.view_radius + 1]
 
-        obs_map_state = np.zeros((2*self.view_radius +1, 2*self.view_radius + 1, 2))
+        obs_map_state = np.zeros((2*self.view_radius + 1, 2*self.view_radius + 1, 2))
 
-        obs_other_agents_state = np.zeros((2*self.view_radius +1, 2*self.view_radius +1, 4))
+        obs_other_agents_state = np.zeros((2*self.view_radius + 1, 2*self.view_radius + 1, 4))
 
         def relative_pos(pos):
             return [agent.position[0] - pos[0], agent.position[1] - pos[1]]