diff --git a/src/Scheduler/SchedulerListenerService.php b/src/Scheduler/SchedulerListenerService.php
index 5a37555..e962cd1 100644
--- a/src/Scheduler/SchedulerListenerService.php
+++ b/src/Scheduler/SchedulerListenerService.php
@@ -52,7 +52,7 @@ public function __construct(string $entryPointName, SchedulerProvider $scheduler
      * @Annotations\Services\CommandHandler()
      *
      * @param EmitSchedulerOperationCommand $command
-     * @param TestApplicationContext        $context
+     * @param ExecutionContextInterface     $context
      * @param SchedulerProvider             $provider
      *
      * @return void
@@ -66,7 +66,7 @@ public function __construct(string $entryPointName, SchedulerProvider $scheduler
      */
     public function handleEmitSchedulerOperationCommand(
         EmitSchedulerOperationCommand $command,
-        TestApplicationContext $context,
+        ExecutionContextInterface $context,
         SchedulerProvider $provider
     ): void
     {