diff --git a/queue_job/controllers/main.py b/queue_job/controllers/main.py
index 0338c823c5..a93c644841 100644
--- a/queue_job/controllers/main.py
+++ b/queue_job/controllers/main.py
@@ -36,7 +36,7 @@ def _try_perform_job(self, env, job):
         job.perform()
         # Triggers any stored computed fields before calling 'set_done'
         # so that will be part of the 'exec_time'
-        env["base"].flush()
+        env.flush_all()
         job.set_done()
         job.store()
         env.flush_all()