diff --git a/setup.cfg b/setup.cfg
index f1b61bd..1c3387a 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -29,7 +29,7 @@ include_package_data = True
 packages = find:
 python_requires = >=3.9.0, <4
 install_requires =
-    websockets >= 9.1, < 12.0
+    websockets >= 9.1
     construct >= 2.9.0, < 3.0.0
     aiohttp >= 3.7.0, < 4.0.0
 tests_require =
diff --git a/tests/conftest.py b/tests/conftest.py
index 7469c71..185ad98 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,7 +1,7 @@
 from unittest.mock import AsyncMock, patch
 
 import pytest
-from websockets.legacy.protocol import WebSocketCommonProtocol
+from websockets.client import WebSocketClientProtocol
 
 from vallox_websocket_api import Client, Vallox
 
@@ -25,7 +25,7 @@ async def vallox():
 @pytest.fixture
 def ws():
     with patch("websockets.client.connect") as connect:
-        protocol_mock = AsyncMock(spec=WebSocketCommonProtocol)
+        protocol_mock = AsyncMock(spec=WebSocketClientProtocol)
         connect.return_value.__aenter__.side_effect = protocol_mock
 
         yield protocol_mock.return_value
diff --git a/vallox_websocket_api/__init__.py b/vallox_websocket_api/__init__.py
index dfe0d7a..d6eff15 100644
--- a/vallox_websocket_api/__init__.py
+++ b/vallox_websocket_api/__init__.py
@@ -20,4 +20,4 @@
     "ValloxWebsocketException",
 ]
 
-__version__ = "5.1.0"
+__version__ = "5.1.1"