diff --git a/classes/doi/Repository.php b/classes/doi/Repository.php
index 932842e0003..706fc029993 100644
--- a/classes/doi/Repository.php
+++ b/classes/doi/Repository.php
@@ -260,8 +260,8 @@ public function isAssigned(int $doiId, string $pubObjectType): bool
 
             return Repo::submissionFile()
                 ->getCollector()
-                ->getIds()
-                ->count();
+                ->getQueryBuilder()
+                ->safeCount();
         };
 
         $isAssigned = match ($pubObjectType) {