diff --git a/tests/web_socket_handler_test.py b/tests/web_socket_handler_test.py
index ac09df8e..ee2fe2e9 100644
--- a/tests/web_socket_handler_test.py
+++ b/tests/web_socket_handler_test.py
@@ -1,7 +1,7 @@
 import json
 from unittest.mock import Mock, patch
 import pytest
-from src.pieces.api import WebSocketManager
+from pieces.api import WebSocketManager
 
 class TestWebSocketManager: