diff --git a/pyproject.toml b/pyproject.toml
index 8b556ff..268f2cf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zai-sdk"
-version = "0.0.3.1"
+version = "0.0.3.2"
 description = "A SDK library for accessing big model apis from Z.ai"
 authors = ["Z.ai"]
 readme = "README.md"
diff --git a/src/zai/_version.py b/src/zai/_version.py
index f38c874..a1e3f00 100644
--- a/src/zai/_version.py
+++ b/src/zai/_version.py
@@ -1,2 +1,2 @@
 __title__ = 'Z.ai'
-__version__ = '0.0.3.1'
+__version__ = '0.0.3.2'
diff --git a/src/zai/api_resource/batch/batches.py b/src/zai/api_resource/batch/batches.py
index a4d3ee7..70fccfb 100644
--- a/src/zai/api_resource/batch/batches.py
+++ b/src/zai/api_resource/batch/batches.py
@@ -17,7 +17,7 @@
 from zai.types.batch import Batch, BatchCreateParams, BatchListParams
 
 if TYPE_CHECKING:
-	from .._client import ZaiClient
+	from zai._client import ZaiClient
 
 
 class Batches(BaseAPI):
diff --git a/src/zai/api_resource/files/files.py b/src/zai/api_resource/files/files.py
index 3d89c37..c613f73 100644
--- a/src/zai/api_resource/files/files.py
+++ b/src/zai/api_resource/files/files.py
@@ -28,7 +28,7 @@
 )
 
 if TYPE_CHECKING:
-	from .._client import ZaiClient
+	from zai._client import ZaiClient
 
 
 class Files(BaseAPI):