diff --git a/src/k8s-extension/azext_k8s_extension/custom.py b/src/k8s-extension/azext_k8s_extension/custom.py
index c0a2c4cdb6a..abff15a0246 100644
--- a/src/k8s-extension/azext_k8s_extension/custom.py
+++ b/src/k8s-extension/azext_k8s_extension/custom.py
@@ -5,7 +5,7 @@
 
 # pylint: disable=unused-argument,too-many-locals
 
-from azext_k8s_extension.utils import (
+from .utils import (
     get_cluster_rp_api_version,
     is_dogfood_cluster,
     read_config_settings_file,
diff --git a/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py b/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py
index 793691cf4af..6334877f59e 100644
--- a/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py
+++ b/src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py
@@ -10,7 +10,7 @@
 import copy
 from hashlib import md5
 from typing import Any, Dict, List, Tuple
-from azext_k8s_extension.utils import get_cluster_rp_api_version
+from ..utils import get_cluster_rp_api_version
 
 import azure.mgmt.relay
 import azure.mgmt.relay.models
diff --git a/src/k8s-extension/azext_k8s_extension/utils.py b/src/k8s-extension/azext_k8s_extension/utils.py
index 9c7322cdf70..c587c3050a1 100644
--- a/src/k8s-extension/azext_k8s_extension/utils.py
+++ b/src/k8s-extension/azext_k8s_extension/utils.py
@@ -6,7 +6,7 @@
 import json
 from typing import Tuple
 from urllib.parse import urlparse
-from azext_k8s_extension import consts
+from . import consts
 from azure.cli.core.azclierror import InvalidArgumentValueError