diff --git a/__init___.py b/__init___.py
new file mode 100644
index 0000000..4b6fe15
--- /dev/null
+++ b/__init___.py
@@ -0,0 +1,2 @@
+# repo-root/__init__.py
+from .torch_compile_lora_safe import NODE_CLASS_MAPPINGS
diff --git a/lora_safe_compile/__init___.py b/lora_safe_compile/__init___.py
deleted file mode 100644
index 70699f2..0000000
--- a/lora_safe_compile/__init___.py
+++ /dev/null
@@ -1 +0,0 @@
-from .torch_compile_lora_safe import NODE_CLASS_MAPPINGS
\ No newline at end of file
diff --git a/lora_safe_compile/torch_compile_lora_safe.py b/torch_compile_lora_safe.py
similarity index 100%
rename from lora_safe_compile/torch_compile_lora_safe.py
rename to torch_compile_lora_safe.py