diff --git a/InvenTree/common/test_notifications.py b/InvenTree/common/test_notifications.py
index 30d4713a3e..6eec11e056 100644
--- a/InvenTree/common/test_notifications.py
+++ b/InvenTree/common/test_notifications.py
@@ -77,7 +77,7 @@ class BaseNotificationTests(BaseNotificationIntegrationTest):
             def send(self, target):
                 raise KeyError('This could be any error')
 
-        self._notification_run()
+        self._notification_run(ErrorImplementation)
 
 
 class BulkNotificationMethodTests(BaseNotificationIntegrationTest):