diff --git a/air_llm/__init__.py b/air_llm/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/air_llm/tests/__init__.py b/air_llm/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/air_llm/tests/test_automodel.py b/air_llm/tests/test_automodel.py
index 8224295..1b71148 100644
--- a/air_llm/tests/test_automodel.py
+++ b/air_llm/tests/test_automodel.py
@@ -1,10 +1,9 @@
 import sys
 import unittest
 
-import torch
-sys.path.insert(0, '../airllm')
+#sys.path.insert(0, '../airllm')
 
-from auto_model import AutoModel
+from ..airllm.auto_model import AutoModel