diff --git a/test/WsPrivateTest.py b/test/WsPrivateTest.py
index 2a8f0a0..cf2ab5e 100644
--- a/test/WsPrivateTest.py
+++ b/test/WsPrivateTest.py
@@ -1,6 +1,6 @@
 import time
 
-from WsPrivate import WsPrivate
+from okx.websocket.WsPrivate import WsPrivate
 
 
 def privateCallback(message):