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