diff --git a/src/k8s-extension/HISTORY_private.rst b/src/k8s-extension/HISTORY_private.rst
index 3d3fd9873f1..ab866455ee1 100644
--- a/src/k8s-extension/HISTORY_private.rst
+++ b/src/k8s-extension/HISTORY_private.rst
@@ -3,6 +3,11 @@
 Release History
 ===============
 
+0.4.0-beta.2
+++++++++++++++++++
+
+* Fix import bug
+
 0.4.0-beta.1
 ++++++++++++++++++
 
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 e2cce49e8eb..e18b8ab455c 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
diff --git a/src/k8s-extension/setup_private.py b/src/k8s-extension/setup_private.py
index 688a9ad761a..bcf133aae62 100644
--- a/src/k8s-extension/setup_private.py
+++ b/src/k8s-extension/setup_private.py
@@ -32,7 +32,7 @@
 # TODO: Add any additional SDK dependencies here
 DEPENDENCIES = []
 
-VERSION = "0.4.0-beta.1"
+VERSION = "0.4.0-beta.2"
 
 with open('README.rst', 'r', encoding='utf-8') as f:
     README = f.read()