1 files changed, 1 insertions, 2 deletions

diff --git a/tests/handlers/test_saml.py b/tests/handlers/test_saml.py

index 30efd43b40..8cfc184fef 100644
--- a/tests/handlers/test_saml.py
+++ b/tests/handlers/test_saml.py
@@ -13,8 +13,7 @@
 #  limitations under the License.
 
 from typing import Optional
-
-from mock import Mock
+from unittest.mock import Mock
 
 import attr