diff --git a/tests/worker/test_update_with_start.py b/tests/worker/test_update_with_start.py
index 3f975742..788cb4b6 100644
--- a/tests/worker/test_update_with_start.py
+++ b/tests/worker/test_update_with_start.py
@@ -1,12 +1,10 @@
 from __future__ import annotations
 
 import uuid
-from contextlib import contextmanager
 from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
-from typing import Any, Iterator
-from unittest.mock import patch
+from typing import Any
 
 import pytest
 
@@ -319,19 +317,6 @@ async def _do_start_update_test(
             )
             assert await update_handle.result() == "update-result-1"
 
-    @contextmanager
-    def assert_network_call(
-        self,
-        expect_network_call: bool,
-    ) -> Iterator[None]:
-        with patch.object(
-            self.client.workflow_service,
-            "poll_workflow_execution_update",
-            wraps=self.client.workflow_service.poll_workflow_execution_update,
-        ) as _wrapped_poll:
-            yield
-            assert _wrapped_poll.called == expect_network_call
-
 
 async def test_update_with_start_sets_first_execution_run_id(
     client: Client, env: WorkflowEnvironment