diff --git a/torch_training/multi_agent_inference.py b/torch_training/multi_agent_inference.py
index cb8fa32397f668288e7dd689ea68676ed1cc9592..b235e9d34d698a7dbb05f51884fe20b638cbdc20 100644
--- a/torch_training/multi_agent_inference.py
+++ b/torch_training/multi_agent_inference.py
@@ -37,7 +37,7 @@ n_agents = 10
 observation_builder = TreeObsForRailEnv(max_depth=2)
 
 # Use a the malfunction generator to break agents from time to time
-stochastic_data = MalfunctionParameters(malfunction_rate=10000,  # Rate of malfunction occurence
+stochastic_data = MalfunctionParameters(malfunction_rate=1./10000,  # Rate of malfunction occurence
                                         min_duration=15,  # Minimal duration of malfunction
                                         max_duration=50  # Max duration of malfunction
                                         )