diff --git a/hbmqtt/client.py b/hbmqtt/client.py
index 3a9b2d5..dbf47e3 100644
--- a/hbmqtt/client.py
+++ b/hbmqtt/client.py
@@ -18,7 +18,7 @@ from hbmqtt.mqtt.protocol.handler import ProtocolHandlerException
 from hbmqtt.mqtt.constants import QOS_0, QOS_1, QOS_2
 import websockets
 from websockets.uri import InvalidURI
-from websockets.handshake import InvalidHandshake
+from websockets.exceptions import InvalidHandshake
 from collections import deque
 import sys
 if sys.version_info < (3, 5):