diff --git a/markdown_view_plugin/hatch_build.py b/markdown_view_plugin/hatch_build.py
index f023f9d..27e27c6 100644
--- a/markdown_view_plugin/hatch_build.py
+++ b/markdown_view_plugin/hatch_build.py
@@ -21,4 +21,4 @@ class MarkdownBuildHook(BuildHookInterface):
         subprocess.run(["pnpm", "run", "build"], cwd=ui_dir, check=True)
 
         # 2. Force-include the compiled assets in the wheel
-        build_data["force_include"][str(dist_dir)] = "ui/dist"
+        build_data["force_include"][str(dist_dir)] = "markdown_view_plugin/ui/dist"
diff --git a/markdown_view_plugin/markdown_view_plugin/__init__.py b/markdown_view_plugin/markdown_view_plugin/__init__.py
index e69de29..548a90d 100644
--- a/markdown_view_plugin/markdown_view_plugin/__init__.py
+++ b/markdown_view_plugin/markdown_view_plugin/__init__.py
@@ -0,0 +1,3 @@
+from markdown_view_plugin.markdown_view_plugin import MarkdownViewPlugin
+
+__all__ = ["MarkdownViewPlugin"]
diff --git a/markdown_view_plugin/pyproject.toml b/markdown_view_plugin/pyproject.toml
index 4740b39..1f30d48 100644
--- a/markdown_view_plugin/pyproject.toml
+++ b/markdown_view_plugin/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "markdown-view-plugin"
-version = "0.1.1"
+version = "0.1.8"
 description = "Airflow UI plugin to render Markdown content using FastAPI and React."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -29,8 +29,6 @@ classifiers = [
 license = {text = "MIT"} # Updated to new license expression
 # license-files = ["LICEN[CS]E*"] # Optional: if you add a LICENSE file
 
-# Hatch-specific way to specify top-level Python modules
-py-modules = ["markdown_view_plugin"]
 
 [project.urls]
 GitHub = "https://github.com/abhishekbhakat/airflow-markdown-view-plugin"  # Replace with actual URL if it exists
@@ -56,13 +54,29 @@ build-backend = "hatchling.build"
 path = "hatch_build.py"
 
 [tool.hatch.build.targets.wheel]
-# Ensure data files are included in the wheel at the root.
-# markdown_view_plugin.py will be handled by `project.py-modules`.
-# These paths are relative to the pyproject.toml file.
 include = [
+    "markdown_view_plugin",
     "view.md",
     "README.md",
-    "ui/dist"  # This tells Hatch to include the ui/dist directory and its contents
+]
+exclude = [
+    "markdown_view_plugin/markdown_view_plugin/ui/*",
+    "markdown_view_plugin/markdown_view_plugin/ui/**",
+    "!markdown_view_plugin/markdown_view_plugin/ui/dist",
+    "!markdown_view_plugin/markdown_view_plugin/ui/dist/**"
+]
+
+[tool.hatch.build.targets.sdist]
+include = [
+    "markdown_view_plugin",
+    "view.md",
+    "README.md",
+]
+exclude = [
+    "markdown_view_plugin/markdown_view_plugin/ui/*",
+    "markdown_view_plugin/markdown_view_plugin/ui/**",
+    "!markdown_view_plugin/markdown_view_plugin/ui/dist",
+    "!markdown_view_plugin/markdown_view_plugin/ui/dist/**"
 ]
 
 [tool.ruff]