diff --git a/WebSocketServer.php b/WebSocketServer.php
index a62ca92..07f7d49 100644
--- a/WebSocketServer.php
+++ b/WebSocketServer.php
@@ -17,7 +17,8 @@
 use consik\yii2websocket\events\WSClientMessageEvent;
 use Ratchet\ConnectionInterface;
 use Ratchet\Http\HttpServer;
-use Ratchet\MessageComponentInterface;
+use Ratchet\RFC6455\Messaging\MessageInterface;
+use Ratchet\WebSocket\MessageComponentInterface;
 use Ratchet\Server\IoServer;
 use Ratchet\WebSocket\WsServer;
 use Symfony\Component\Console\Event\ConsoleEvent;
@@ -187,8 +188,10 @@ function onError(ConnectionInterface $conn, \Exception $e)
      * @event WSClientCommandEvent EVENT_CLIENT_RUN_COMMAND
      * @event WSClientCommandEvent EVENT_CLIENT_END_COMMAND
      */
-    function onMessage(ConnectionInterface $from, $msg)
+    function onMessage(ConnectionInterface $from, MessageInterface $msg)
     {
+        $msg = $msg->getPayload();
+
         $this->trigger(self::EVENT_CLIENT_MESSAGE, new WSClientMessageEvent([
             'client' => $from,
             'message' => $msg
diff --git a/composer.json b/composer.json
index 7afa616..bde2213 100644
--- a/composer.json
+++ b/composer.json
@@ -12,7 +12,7 @@
 	],
 	"require": {
 		"yiisoft/yii2": "*",
-		"cboden/ratchet": "0.3.*"
+		"cboden/ratchet": "0.4.*"
 	},
 	"autoload": {
 		"psr-4": {