diff --git a/dist/flatland_rl-3.0.0rc2-py2.py3-none-any.whl b/dist/flatland_rl-3.0.0rc2-py2.py3-none-any.whl
index 9cd26e097480396770d3086f0a54cd158284a2d1..4657ff6303ca9fcbc6fb21a1185285ef96e81f0b 100644
Binary files a/dist/flatland_rl-3.0.0rc2-py2.py3-none-any.whl and b/dist/flatland_rl-3.0.0rc2-py2.py3-none-any.whl differ
diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py
index c1d3d446e55d0b68dd6e61b57710177dce9fdf63..97dce409039a800bfc1d3f38898b6d8de207f4f5 100644
--- a/flatland/evaluators/service.py
+++ b/flatland/evaluators/service.py
@@ -867,7 +867,7 @@ class FlatlandRemoteEvaluationService:
             ))
 
         # We count the number of agents that malfunctioned by checking how many have 1 more steps left before recovery
-        num_malfunctioning = sum(agent.malfunction_data['malfunction'] == 1 for agent in self.env.agents)
+        num_malfunctioning = sum(agent.malfunction_handler._malfunction_down_counter == 1 for agent in self.env.agents)
 
         if self.verbose and num_malfunctioning > 0:
             print("Step {}: {} agents have malfunctioned and will recover next step".format(self.current_step, num_malfunctioning))