1 files changed, 3 insertions, 1 deletions
diff --git a/src/Worker.php b/src/Worker.php
index 3d1888e3..2da16e86 100644
--- a/src/Worker.php
+++ b/src/Worker.php
@@ -11,6 +11,8 @@ namespace Spiral\RoadRunner;
 
 use Spiral\Goridge\Exceptions\GoridgeException;
 use Spiral\Goridge\RelayInterface as Relay;
+use Spiral\Goridge\SocketRelay;
+use Spiral\Goridge\StreamRelay;
 use Spiral\RoadRunner\Exception\RoadRunnerException;
 
 /**
@@ -28,7 +30,7 @@ class Worker
     // Send as response context to request worker termination
     public const STOP = '{"stop":true}';
 
-    /** @var Relay */
+    /** @var Relay|StreamRelay|SocketRelay */
     private $relay;
 
     /** @var bool */