diff --git a/build.rs b/build.rs
index 2fce84c..6b430b5 100644
--- a/build.rs
+++ b/build.rs
@@ -3,7 +3,7 @@ use std::process::Command;
 
 fn main() {
     // Path to your Python script
-    let python_script = "src/kernel.py";
+    let python_script = "src/python/kernel.py";
 
     let python_executable = if Path::new(".venv").exists() {
         if cfg!(windows) {
diff --git a/pyproject.toml b/pyproject.toml
index 0c68c54..a4db7a4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,3 +13,6 @@ dependencies = [
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/python"]
diff --git a/src/kernel.py b/src/python/kernel.py
similarity index 100%
rename from src/kernel.py
rename to src/python/kernel.py