diff --git a/tests/worker/test_workflow.py b/tests/worker/test_workflow.py
index a987d1b34..4ca20eb30 100644
--- a/tests/worker/test_workflow.py
+++ b/tests/worker/test_workflow.py
@@ -3193,7 +3193,10 @@ async def test_workflow_query_rpc_timeout(client: Client):
         )
     assert (
         err.value.status == RPCStatusCode.CANCELLED
-        and "timeout" in str(err.value).lower()
+        and (
+            "timeout" in str(err.value).lower()
+            or "http2 error" in str(err.value).lower()
+        )
     ) or err.value.status == RPCStatusCode.DEADLINE_EXCEEDED