diff --git a/test/WsPublicTest.py b/test/WsPublicTest.py
index 17c106e..e0fbec5 100644
--- a/test/WsPublicTest.py
+++ b/test/WsPublicTest.py
@@ -1,6 +1,6 @@
 import time
 
-from WsPublic import WsPublic
+from okx.websocket.WsPublic import WsPublic
 
 
 def publicCallback(message):