diff --git a/tests/test_source_azure_key_vault.py b/tests/test_source_azure_key_vault.py
index 934d9a5b..05852716 100644
--- a/tests/test_source_azure_key_vault.py
+++ b/tests/test_source_azure_key_vault.py
@@ -3,7 +3,6 @@
 """
 
 import pytest
-from azure.keyvault.secrets import SecretClient
 from pydantic import BaseModel, Field
 from pytest_mock import MockerFixture
 
@@ -19,7 +18,7 @@
     import_azure_key_vault()
     from azure.core.exceptions import ResourceNotFoundError
     from azure.identity import DefaultAzureCredential
-    from azure.keyvault.secrets import KeyVaultSecret, SecretProperties
+    from azure.keyvault.secrets import KeyVaultSecret, SecretClient, SecretProperties
 except ImportError:
     azure_key_vault = False