diff --git a/main.py b/main.py
index ed05316bf898163546382a9c96aa395d84836fa6..995ae335fb2f612553ca0746662a5baa43a87ebb 100644
--- a/main.py
+++ b/main.py
@@ -210,6 +210,7 @@ def main(args, env, agent, buffer, writer, stats:Statistics):
                 stats.total_updates += 1
                 agent.update(buffer=buffer, avg_failures=avg_failures, total_episodes=stats.total_episodes + i)
             stats.end(name="update")
+            agent.after_updates()
             if args.clear_buffer:
                 buffer.clear()