diff --git a/ib_async/ib.py b/ib_async/ib.py
index 3edfb8d..05d43eb 100644
--- a/ib_async/ib.py
+++ b/ib_async/ib.py
@@ -347,7 +347,7 @@ def connect(
         account: str = "",
         raiseSyncErrors: bool = False,
         fetchFields: StartupFetch = StartupFetchALL,
-    ):
+    ) -> 'IB':
         """
         Connect to a running TWS or IB gateway application.
         After the connection is made the client is fully synchronized
@@ -2044,7 +2044,7 @@ async def connectAsync(
         account: str = "",
         raiseSyncErrors: bool = False,
         fetchFields: StartupFetch = StartupFetchALL,
-    ):
+    ) -> 'IB':
         clientId = int(clientId)
         self.wrapper.clientId = clientId
         timeout = timeout or None