diff --git a/src/paperless_remote/tests/test_checks.py b/src/paperless_remote/tests/test_checks.py
index ac65da27f..892cb360d 100644
--- a/src/paperless_remote/tests/test_checks.py
+++ b/src/paperless_remote/tests/test_checks.py
@@ -1,4 +1,5 @@
-from django.test import TestCase
+from unittest import TestCase
+
 from django.test import override_settings
 
 from paperless_remote import check_remote_parser_configured
@@ -18,7 +19,7 @@ class TestChecks(TestCase):
         self.assertEqual(len(msgs), 1)
         self.assertTrue(
             msgs[0].msg.startswith(
-                "Azure AI remote parser requires endpoint to be configured.",
+                "Azure AI remote parser requires endpoint and API key to be configured.",
             ),
         )