diff --git a/tests/test_multi_connect.py b/tests/test_multi_connect.py
index e26bc6a8..2b836515 100644
--- a/tests/test_multi_connect.py
+++ b/tests/test_multi_connect.py
@@ -4,7 +4,7 @@ import time
 
 from mpos import App, AppManager, TaskManager
 
-from websocket import WebSocketApp
+from uaiowebsocket import WebSocketApp
 
 # demo_multiple_ws.py
 import asyncio
diff --git a/tests/test_multi_websocket_with_bad_ones.py b/tests/test_multi_websocket_with_bad_ones.py
index 5fbda5a7..d6811bd0 100644
--- a/tests/test_multi_websocket_with_bad_ones.py
+++ b/tests/test_multi_websocket_with_bad_ones.py
@@ -5,7 +5,7 @@ import time
 from mpos import App, AppManager
 from mpos import TaskManager
 
-from websocket import WebSocketApp
+from uaiowebsocket import WebSocketApp
 
 import asyncio
 import aiohttp
diff --git a/tests/test_websocket.py b/tests/test_websocket.py
index 46ad55af..258cdff5 100644
--- a/tests/test_websocket.py
+++ b/tests/test_websocket.py
@@ -6,7 +6,7 @@ import time
 from mpos import App, AppManager
 from mpos import TaskManager
 
-from websocket import WebSocketApp
+from uaiowebsocket import WebSocketApp
 
 class TestMutlipleWebsocketsAsyncio(unittest.TestCase):