diff --git a/pythonbpf/tbaa_gen/__init__.py b/pythonbpf/tbaa_gen/__init__.py
deleted file mode 100644
index e69de29..0000000
diff --git a/pythonbpf/vmlinux_parser/import_detector.py b/pythonbpf/vmlinux_parser/import_detector.py
index 2ce9cb5..f5789ce 100644
--- a/pythonbpf/vmlinux_parser/import_detector.py
+++ b/pythonbpf/vmlinux_parser/import_detector.py
@@ -5,7 +5,7 @@ import importlib
 import inspect
 
 from .dependency_handler import DependencyHandler
-from .ir_generation import IRGenerator
+from .ir_gen import IRGenerator
 from .class_handler import process_vmlinux_class
 
 logger = logging.getLogger(__name__)
diff --git a/pythonbpf/vmlinux_parser/ir_gen/__init__.py b/pythonbpf/vmlinux_parser/ir_gen/__init__.py
new file mode 100644
index 0000000..3a13651
--- /dev/null
+++ b/pythonbpf/vmlinux_parser/ir_gen/__init__.py
@@ -0,0 +1,3 @@
+from .ir_generation import IRGenerator
+
+__all__ = ["IRGenerator"]
diff --git a/pythonbpf/vmlinux_parser/ir_generation.py b/pythonbpf/vmlinux_parser/ir_gen/ir_generation.py
similarity index 85%
rename from pythonbpf/vmlinux_parser/ir_generation.py
rename to pythonbpf/vmlinux_parser/ir_gen/ir_generation.py
index c66ba11..e4dae7d 100644
--- a/pythonbpf/vmlinux_parser/ir_generation.py
+++ b/pythonbpf/vmlinux_parser/ir_gen/ir_generation.py
@@ -1,5 +1,5 @@
 import logging
-from .dependency_handler import DependencyHandler
+from pythonbpf.vmlinux_parser.dependency_handler import DependencyHandler
 
 logger = logging.getLogger(__name__)