diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index ba81e7a1..51323ed4 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -12,7 +12,7 @@ on:
 
 jobs:
   check:
-    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@ruff
+    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@main
 
     strategy:
       fail-fast: false
@@ -28,7 +28,7 @@ jobs:
   # Checks the library using minimum version resolution
   # `uv` has this feature built-in, c.f. https://github.com/astral-sh/uv
   check-min-version:
-    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@ruff
+    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@main
 
     with:
       python-version: "3.10"
diff --git a/geoengine/workflow.py b/geoengine/workflow.py
index a4b8c27f..579b8adb 100644
--- a/geoengine/workflow.py
+++ b/geoengine/workflow.py
@@ -25,6 +25,7 @@
 import requests as req
 import rioxarray
 import websockets
+import websockets.asyncio.client
 import xarray as xr
 from owslib.util import Authentication, ResponseWrapper
 from owslib.wcs import WebCoverageService
@@ -622,7 +623,7 @@ async def raster_stream(
 
         async with websockets.asyncio.client.connect(
             uri=self.__replace_http_with_ws(url),
-            extra_headers=session.auth_header,
+            additional_headers=session.auth_header,
             open_timeout=open_timeout,
             max_size=None,
         ) as websocket:
diff --git a/pyproject.toml b/pyproject.toml
index b02362ad..6e271f9b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ dependencies = [
     "rioxarray >=0.9.1, <0.20",
     "StrEnum >=0.4.6,<0.5",               # TODO: use from stdlib when `python_requires = >=3.11`
     "vega >= 3.5,<4.2",
-    "websockets >= 14.0,<16",
+    "websockets >= 14.2,<16",
     "xarray >=0.19,<2025.8",
     "urllib3 >= 2.1, < 2.6",
     "pydantic >= 2.10.6, < 2.12",