diff --git a/air_llm/tests/test_automodel.py b/air_llm/tests/test_automodel.py
index 493af23..b546161 100644
--- a/air_llm/tests/test_automodel.py
+++ b/air_llm/tests/test_automodel.py
@@ -2,7 +2,7 @@ import sys
 import unittest
 
 import torch
-#sys.path.insert(0, '../airllm')
+#sys.path.insert(0, '..')
 
 from airllm.auto_model import AutoModel