diff --git a/arkindex_worker/worker/base.py b/arkindex_worker/worker/base.py
index 37da3a566ae9ffa01b1e4767fb5b863a3369facf..1be5df4f422b63e49f83ad06a40a06eb0a81e99e 100644
--- a/arkindex_worker/worker/base.py
+++ b/arkindex_worker/worker/base.py
@@ -266,7 +266,7 @@ class BaseWorker:
         model_version = worker_run.get("model_version")
         if model_version:
             logger.info("Loaded model version configuration from WorkerRun")
-            self.model_configuration.update(model_version.get("configuration", {}))
+            self.model_configuration.update(model_version["configuration"])
 
             # Set model_version ID as worker attribute
             self.model_version_id = model_version["id"]