diff --git a/src/Fabiang/Xmpp/Connection/Socket.php b/src/Fabiang/Xmpp/Connection/Socket.php
index b1167ef..e873bd6 100644
--- a/src/Fabiang/Xmpp/Connection/Socket.php
+++ b/src/Fabiang/Xmpp/Connection/Socket.php
@@ -59,7 +59,7 @@ class Socket extends AbstractConnection implements SocketConnectionInterface
     /**
      * Socket.
      *
-     * @var Socket
+     * @var SocketClient
      */
     protected $socket;
 
@@ -167,7 +167,7 @@ protected function getAddress()
     /**
      * Return socket instance.
      *
-     * @return StreamSocket
+     * @return SocketClient
      */
     public function getSocket()
     {
diff --git a/src/Fabiang/Xmpp/Connection/SocketConnectionInterface.php b/src/Fabiang/Xmpp/Connection/SocketConnectionInterface.php
index 456c5d8..ea17ef8 100644
--- a/src/Fabiang/Xmpp/Connection/SocketConnectionInterface.php
+++ b/src/Fabiang/Xmpp/Connection/SocketConnectionInterface.php
@@ -49,7 +49,7 @@ interface SocketConnectionInterface
     /**
      * Set socket instance.
      *
-     * @param \Socket\Raw\Socket $socket
+     * @param SocketClient $socket
      * @return $this
      */
     public function setSocket(SocketClient $socket);