diff --git a/api/controller/image.py b/api/controller/image.py
index 1fdbade..5546a51 100644
--- a/api/controller/image.py
+++ b/api/controller/image.py
@@ -49,8 +49,8 @@
     constants.IMAGE_TYPE_TEST: _TEST_ID,
     _RECOVERY_ID: constants.IMAGE_TYPE_RECOVERY,
     constants.IMAGE_TYPE_RECOVERY: _RECOVERY_ID,
-    _FACTORY_ID: constants.IMAGE_TYPE_FACTORY,
-    constants.IMAGE_TYPE_FACTORY: _FACTORY_ID,
+    _FACTORY_ID: constants.IMAGE_TYPE_FACTORY_SHIM,
+    constants.IMAGE_TYPE_FACTORY_SHIM: _FACTORY_ID,
     _FIRMWARE_ID: constants.IMAGE_TYPE_FIRMWARE,
     constants.IMAGE_TYPE_FIRMWARE: _FIRMWARE_ID,
 }