diff --git a/scripts/websocket_test.py b/scripts/websocket_test.py
index 21c7f16..fce9cd7 100755
--- a/scripts/websocket_test.py
+++ b/scripts/websocket_test.py
@@ -45,10 +45,10 @@
     authbytes = "{}:{}".format(args.username, args.password).encode("ascii")
     auth = "Basic {}".format(base64.b64encode(authbytes).decode("ascii"))
     headers = {"Authorization": auth}
-    async with websockets.connect(
-        uri, ssl=ssl_context, extra_headers=headers
-    ) if args.ssl else websockets.connect(
-        uri, extra_headers=headers
+    async with (
+        websockets.connect(uri, ssl=ssl_context, extra_headers=headers)
+        if args.ssl
+        else websockets.connect(uri, extra_headers=headers)
     ) as websocket:
         request = json.dumps(
             {