1 files changed, 2 insertions, 1 deletions
diff --git a/tests/handlers/test_saml.py b/tests/handlers/test_saml.py
index 4df6a4d029..eac7f74591 100644
--- a/tests/handlers/test_saml.py
+++ b/tests/handlers/test_saml.py
@@ -18,7 +18,6 @@ from unittest.mock import Mock
 import attr
 
 from synapse.api.errors import RedirectException
-from synapse.handlers.saml import load_default_or_legacy_saml2_mapping_provider
 
 from tests.test_utils import simple_async_mock
 from tests.unittest import HomeserverTestCase, override_config
@@ -28,6 +27,8 @@ try:
     import saml2.config
     from saml2.sigver import SigverError
 
+    from synapse.handlers.saml import load_default_or_legacy_saml2_mapping_provider
+
     has_saml2 = True
 
     # pysaml2 can be installed and imported, but might not be able to find xmlsec1.