diff --git a/nostr/relay.py b/nostr/relay.py
index 01ac875..80fca7e 100644
--- a/nostr/relay.py
+++ b/nostr/relay.py
@@ -30,9 +30,9 @@ class Relay:
         self.ws = ws
         self.subscriptions = subscriptions
 
-    def open_websocket_connection(self, timeout=None) -> None:
+    def open_websocket_connection(self, timeout: int=None) -> None:
         if timeout != None:
-            self.ws.connect(self.url, timeout)
+            self.ws.connect(self.url, timeout=timeout)
         else:
             self.ws.connect(self.url)