diff --git a/flatland/envs/rail_env.py b/flatland/envs/rail_env.py
index a61ef02207174d04489b5311dc042b7c06db1412..62efbdc5bc0781c4c7482412dafd98710ed9d14e 100644
--- a/flatland/envs/rail_env.py
+++ b/flatland/envs/rail_env.py
@@ -271,8 +271,7 @@ class RailEnv(Environment):
         agent.malfunction_data['next_malfunction'] -= 1
 
         # Only agents that have a positive rate for malfunctions and are not currently broken are considered
-        if agent.malfunction_data['malfunction_rate'] > 0 >= agent.malfunction_data[
-            'malfunction']:
+        if agent.malfunction_data['malfunction_rate'] > 0 >= agent.malfunction_data['malfunction']:
 
             # If counter has come to zero --> Agent has malfunction
             # set next malfunction time and duration of current malfunction
diff --git a/flatland/envs/rail_generators.py b/flatland/envs/rail_generators.py
index 8f2c5231232c3cc17f50de4b89b88f1c9fdc5d60..40ec2e0df89a3de48bf0a2a4430de3e30fa556e5 100644
--- a/flatland/envs/rail_generators.py
+++ b/flatland/envs/rail_generators.py
@@ -683,8 +683,10 @@ def sparse_rail_generator(num_cities=5, num_intersections=4, num_trainstations=2
                                     0,
                                     width - 1)
                 tries = 0
-                while (station_x, station_y) in train_stations or (station_x, station_y) == node_positions[
-                    trainstation_node] or rail_array[(station_x, station_y)] != 0:
+                while (station_x, station_y) in train_stations \
+                    or (station_x, station_y) == node_positions[trainstation_node] \
+                    or rail_array[(station_x, station_y)] != 0:  # noqa: E125
+
                     station_x = np.clip(
                         node_positions[trainstation_node][0] + np.random.randint(-node_radius, node_radius),
                         0,