diff --git a/setup.py b/setup.py
index f793eb2..b1d767c 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,7 @@ setup(
     packages=find_packages(),
     install_requires=[
         "requests",
-        "websockets==11.0.2",
+        "websockets>=13.1",
         "aiohttp>=3.8",
         "human-readable-ids==0.1.3",
         "jsonpickle==3.0.1",
diff --git a/test.py b/test.py
new file mode 100644
index 0000000..6254378
--- /dev/null
+++ b/test.py
@@ -0,0 +1,16 @@
+import asyncio
+import json
+from streamdeckapi import SDWebsocketMessage, StreamDeckApi
+
+async def __main__():
+    deck = StreamDeckApi("localhost")
+    info = await deck.get_info(False)
+
+    if info is None:
+        print("Error getting info")
+        return
+
+    print(json.dumps(info))
+
+loop = asyncio.get_event_loop()
+loop.run_until_complete(__main__())
diff --git a/test.sh b/test.sh
new file mode 100755
index 0000000..d386dd9
--- /dev/null
+++ b/test.sh
@@ -0,0 +1,9 @@
+#
+# Only used for development!
+#
+
+# Start a server
+#docker run -v /dev/hidraw7:/dev/hidraw7 -p 6153:6153 --privileged --detached ghcr.io/patrick762/streamdeckapi:main
+
+# Run python tests
+python test.py