diff --git a/src/Controller/WorkflowControllerTrait.php b/src/Controller/WorkflowControllerTrait.php
index 813aab3..72e5ea8 100644
--- a/src/Controller/WorkflowControllerTrait.php
+++ b/src/Controller/WorkflowControllerTrait.php
@@ -17,6 +17,7 @@
 use Symfony\Component\Workflow\Exception\LogicException;
 use Symfony\Component\Workflow\Registry;
 use Symfony\Component\Workflow\Workflow;
+use Symfony\Component\Workflow\WorkflowInterface;
 
 /**
  *
@@ -131,7 +132,7 @@ public function workflowApplyTransitionAction(Request $request): Response
     /**
      * @throws InvalidArgumentException
      */
-    protected function getWorkflow(object $object): Workflow
+    protected function getWorkflow(object $object): WorkflowInterface
     {
         $registry = $this->workflowRegistry ?? null;
         if ($registry === null) {