diff --git a/src/Client/WorkflowClientInterface.php b/src/Client/WorkflowClientInterface.php
index f72c2922..6c01a4ba 100644
--- a/src/Client/WorkflowClientInterface.php
+++ b/src/Client/WorkflowClientInterface.php
@@ -93,7 +93,7 @@ public function newUntypedWorkflowStub(
      * Returns workflow stub associated with running workflow.
      *
      * @psalm-template T of object
-     * @param string $class
+     * @param class-string<T> $class
      * @param string $workflowID
      * @param string|null $runID
      * @return T