diff --git a/classes/publication/Repository.php b/classes/publication/Repository.php
index 375cb50261..0b30147ce7 100644
--- a/classes/publication/Repository.php
+++ b/classes/publication/Repository.php
@@ -197,7 +197,7 @@ public function canCurrentUserPublish(int $submissionId, ?User $user = null): bo
         // Check if current user is an author
         $isAuthor = false;
 
-        $submitterAssignments = StageAssignment::withSubmissionId($submissionId)
+        $submitterAssignments = StageAssignment::withSubmissionIds([$submissionId])
             ->withRoleIds([Role::ROLE_ID_AUTHOR])
             ->get();
 
diff --git a/controllers/tab/workflow/WorkflowTabHandler.php b/controllers/tab/workflow/WorkflowTabHandler.php
index dc11dd2fe2..49eec88d2f 100644
--- a/controllers/tab/workflow/WorkflowTabHandler.php
+++ b/controllers/tab/workflow/WorkflowTabHandler.php
@@ -44,7 +44,7 @@ public function fetchTab($args, $request)
                 $errors = [];
                 $context = $request->getContext();
 
-                $submitterAssignments = StageAssignment::withSubmissionId($submission->getId())
+                $submitterAssignments = StageAssignment::withSubmissionIds([$submission->getId()])
                     ->withRoleIds([Role::ROLE_ID_AUTHOR])
                     ->get();