diff --git a/azext_edge/tests/conftest.py b/azext_edge/tests/conftest.py
index 8db86c31a..8d9fa5f3d 100644
--- a/azext_edge/tests/conftest.py
+++ b/azext_edge/tests/conftest.py
@@ -32,6 +32,7 @@ def mocked_azcli_cred_get_token(mocker):
         "azure.identity._credentials.azure_cli.AzureCliCredential.get_token",
     )
     type(patched()).expires_on = PropertyMock(return_value=sys.maxsize)
+    type(patched()).refresh_on = PropertyMock(return_value=sys.maxsize)
     yield patched
 
 
diff --git a/setup.py b/setup.py
index 844b89d1e..eda9f4875 100644
--- a/setup.py
+++ b/setup.py
@@ -30,7 +30,7 @@
 DEPENDENCIES = [
     "rich>=13.6,<14.0",
     "kubernetes>=27.2,<29.0",
-    "azure-identity>=1.14.1,<2.0",
+    "azure-identity>=1.14.1,<1.18.0",
     "protobuf~=4.25.0",
     "opentelemetry-proto~=1.20.0",
     "packaging>=23.2",