diff --git a/src/PipelineRunnerInterface.php b/src/PipelineRunnerInterface.php
index b652778..2b48d06 100644
--- a/src/PipelineRunnerInterface.php
+++ b/src/PipelineRunnerInterface.php
@@ -20,7 +20,7 @@ interface PipelineRunnerInterface
     public function run(
         \Iterator $source,
         \Generator $async,
-        RejectionInterface $rejection,
-        StateInterface $state,
+        StepRejectionInterface $rejection,
+        StepStateInterface $state,
     ): \Iterator;
 }