diff --git a/src/k8s-extension/azext_k8s_extension/_validators.py b/src/k8s-extension/azext_k8s_extension/_validators.py
index 9941c4a117b..31078a4ddbb 100644
--- a/src/k8s-extension/azext_k8s_extension/_validators.py
+++ b/src/k8s-extension/azext_k8s_extension/_validators.py
@@ -4,7 +4,7 @@
 # --------------------------------------------------------------------------------------------
 
 from knack.log import get_logger
-from azext_k8s_extension._client_factory import _resource_providers_client
+from ._client_factory import _resource_providers_client
 from . import consts
 
 
diff --git a/src/k8s-extension/azext_k8s_extension/custom.py b/src/k8s-extension/azext_k8s_extension/custom.py
index f4f2e1cafdd..64c99f25abd 100644
--- a/src/k8s-extension/azext_k8s_extension/custom.py
+++ b/src/k8s-extension/azext_k8s_extension/custom.py
@@ -13,10 +13,8 @@
 from azure.cli.core.azclierror import ResourceNotFoundError, MutuallyExclusiveArgumentError, \
     InvalidArgumentValueError, CommandNotFoundError, RequiredArgumentMissingError
 from azure.cli.core.commands.client_factory import get_subscription_id
-from azext_k8s_extension.vendored_sdks.models import ConfigurationIdentity
-from azext_k8s_extension.vendored_sdks.models import ErrorResponseException
-from azext_k8s_extension.vendored_sdks.models import Scope
-from azext_k8s_extension._validators import validate_cc_registration
+from .vendored_sdks.models import ConfigurationIdentity, ErrorResponseException, Scope
+from ._validators import validate_cc_registration
 
 from .partner_extensions.ContainerInsights import ContainerInsights
 from .partner_extensions.AzureDefender import AzureDefender