diff --git a/tests/test_concurrency_limit.py b/tests/test_concurrency_limit.py
index b4acf90f..958521f4 100644
--- a/tests/test_concurrency_limit.py
+++ b/tests/test_concurrency_limit.py
@@ -13,16 +13,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import TYPE_CHECKING, Any, Dict, List
-
-from sygnal.notifications import ConcurrencyLimitedPushkin, Device, Notification
+from typing import Any, Dict, List
+
+from sygnal.notifications import (
+    ConcurrencyLimitedPushkin,
+    Device,
+    Notification,
+    NotificationContext,
+)
 from sygnal.utils import twisted_sleep
 
 from tests.testutils import TestCase
 
-if TYPE_CHECKING:
-    from sygnal.notifications import NotificationContext
-
 DEVICE_GCM1_EXAMPLE = {
     "app_id": "com.example.gcm",
     "pushkey": "spqrg",
@@ -41,7 +43,7 @@
 
 
 class SlowConcurrencyLimitedDummyPushkin(ConcurrencyLimitedPushkin):
-    async def dispatch_notification(
+    async def _dispatch_notification_unlimited(
         self, n: Notification, device: Device, context: "NotificationContext"
     ) -> List[str]:
         """