diff --git a/tests/test_libs/test_auth.py b/tests/test_libs/test_auth.py
index 7637ab6..2619223 100644
--- a/tests/test_libs/test_auth.py
+++ b/tests/test_libs/test_auth.py
@@ -17,5 +17,5 @@ def fake_session(**kwargs):
 def test_resource(monkeypatch):
     monkeypatch.setattr(config, "load_config_file", fake_config)
     monkeypatch.setattr(boto3, "Session", fake_session)
-    monkeypatch.setattr(auth, "get_endpoint", lambda: None)
+    monkeypatch.setattr(auth, "get_endpoint", lambda url: None)
     assert auth.resource() == "s3_resource"
diff --git a/tests/test_libs/test_gmt.py b/tests/test_libs/test_gmt.py
index 9156606..5e489d7 100644
--- a/tests/test_libs/test_gmt.py
+++ b/tests/test_libs/test_gmt.py
@@ -82,7 +82,7 @@ def fake_auth():
 
 def test_policy_id(monkeypatch):
     monkeypatch.setattr(requests, "get", fake_request)
-    monkeypatch.setattr(auth_lib, "get_endpoint", lambda bucket_name: None)
+    monkeypatch.setattr(auth_lib, "get_endpoint", lambda url, bucket_name: None)
     assert "dd7e84cfe467c0fc11b5b075ac9acd73" == gmt.policy_id(
         "awesome-bucket", fake_auth()
     )