diff --git a/packages/core/upload/server/services/extensions/core/entity-service.js b/packages/core/upload/server/services/extensions/core/entity-service.js
index 7c11d8168b..0fd6904f1d 100644
--- a/packages/core/upload/server/services/extensions/core/entity-service.js
+++ b/packages/core/upload/server/services/extensions/core/entity-service.js
@@ -12,11 +12,11 @@ const addSignedFileUrlsToEntityService = async () => {
   }
 
   const decorator = (service) => ({
-    wrapResult(result) {
+    wrapResult(result, { uid }) {
       if (Array.isArray(result)) {
-        return Promise.all(result.map((entity) => signEntityMedia(entity, service.model.uid)));
+        return Promise.all(result.map((entity) => signEntityMedia(entity, uid)));
       }
-      return signEntityMedia(result, service.model.uid);
+      return signEntityMedia(result, uid);
     },
   });