diff --git a/hatch.toml b/hatch.toml
index 12054f94a..411d3da9b 100644
--- a/hatch.toml
+++ b/hatch.toml
@@ -83,7 +83,7 @@ source_archive = "https://github.com/codegen-sh/codegen-sdk/archive/{commit_hash
 
 [build.targets.wheel.hooks.custom]
 enable-by-default = false
-path = "src/graph_sitter_build/build.py"
+path = "src/codegen/gsbuild/build.py"
 require-runtime-dependencies = true
 [envs.default]
 installer = "uv"
diff --git a/src/graph_sitter_build/build.py b/src/codegen/gsbuild/build.py
similarity index 90%
rename from src/graph_sitter_build/build.py
rename to src/codegen/gsbuild/build.py
index c94f522bf..789d23ef1 100644
--- a/src/graph_sitter_build/build.py
+++ b/src/codegen/gsbuild/build.py
@@ -6,7 +6,7 @@
 
 
 def update_init_file(file: Path) -> None:
-    path = Path(__file__).parent.parent
+    path = Path(__file__).parent.parent.parent
     sys.path.append(str(path))
     from codegen.gscli.generate.runner_imports import generate_exported_modules, get_runner_imports
 
@@ -16,7 +16,7 @@ def update_init_file(file: Path) -> None:
 
 
 class SpecialBuildHook(BuildHookInterface):
-    PLUGIN_NAME = "graph_sitter_build"
+    PLUGIN_NAME = "codegen_build"
 
     def initialize(self, version: str, build_data: dict[str, Any]) -> None:
         file = Path(self.root) / "src" / "codegen" / "sdk" / "__init__.py"