diff --git a/tests/StateDefinitionTypeTest.php b/tests/StateDefinitionTypeTest.php
index 4a8301a..28a5c7e 100644
--- a/tests/StateDefinitionTypeTest.php
+++ b/tests/StateDefinitionTypeTest.php
@@ -4,8 +4,6 @@
 
 use Illuminate\Support\Carbon;
 use Tarfinlabs\EventMachine\Actor\Machine;
-use Tarfinlabs\EventMachine\ContextManager;
-use Tarfinlabs\EventMachine\Behavior\EventBehavior;
 use Tarfinlabs\EventMachine\Enums\StateDefinitionType;
 use Tarfinlabs\EventMachine\Definition\MachineDefinition;
 use Tarfinlabs\EventMachine\Tests\Stubs\Results\GreenResult;
@@ -72,7 +70,7 @@
                 ],
                 'yellow' => [
                     'type'   => 'final',
-                    'result' => function (ContextManager $context, EventBehavior $event): Carbon {
+                    'result' => function (): Carbon {
                         return now();
                     },
                 ],