diff --git a/arkindex_worker/worker/base.py b/arkindex_worker/worker/base.py
index 914389de0f2bbd8f68fd5217ce0ebca0a574bb57..37da3a566ae9ffa01b1e4767fb5b863a3369facf 100644
--- a/arkindex_worker/worker/base.py
+++ b/arkindex_worker/worker/base.py
@@ -269,10 +269,10 @@ class BaseWorker:
             self.model_configuration.update(model_version.get("configuration", {}))
 
             # Set model_version ID as worker attribute
-            self.model_version_id = model_version.get("id")
+            self.model_version_id = model_version["id"]
 
             # Set model details as worker attribute
-            self.model_details = model_version.get("model", {})
+            self.model_details = model_version["model"]
 
         # Retrieve initial configuration from API
         self.config = worker_version["configuration"].get("configuration", {})