diff --git a/rocket_learn/utils/generate_episode.py b/rocket_learn/utils/generate_episode.py
index c73109d..f926c05 100644
--- a/rocket_learn/utils/generate_episode.py
+++ b/rocket_learn/utils/generate_episode.py
@@ -29,9 +29,11 @@ def generate_episode(env: Gym, policies, evaluate=False, scoreboard=None) -> (Li
         if isinstance(env._match._state_setter, DynamicGMSetter):  # noqa
             state_setter = env._match._state_setter.setter  # noqa
             env._match._state_setter.setter = DefaultState()  # noqa
+            env.update_settings(boost_consumption=1)  # remove infinite boost
         else:
             state_setter = env._match._state_setter  # noqa
             env._match._state_setter = DefaultState()  # noqa
+            env.update_settings(boost_consumption=1)  # remove infinite boost
 
         env._match._reward_fn = ConstantReward()  # noqa Save some cpu cycles