diff --git a/funcx_sdk/funcx/sdk/asynchronous/ws_polling_task.py b/funcx_sdk/funcx/sdk/asynchronous/ws_polling_task.py index 825f52b7a..e9932a543 100644 --- a/funcx_sdk/funcx/sdk/asynchronous/ws_polling_task.py +++ b/funcx_sdk/funcx/sdk/asynchronous/ws_polling_task.py @@ -79,7 +79,7 @@ def __init__( async def init_ws(self, start_message_handlers=True): headers = [self.get_auth_header()] try: - self.ws = await websockets.client.connect( + self.ws = await websockets.connect( self.results_ws_uri, extra_headers=headers ) # initial Globus authentication happens during the HTTP portion of the handshake,