diff --git a/llamascript/test.py b/llamascript/test.py
index 74da27f..094cec0 100644
--- a/llamascript/test.py
+++ b/llamascript/test.py
@@ -1,6 +1,6 @@
 import unittest
 from unittest.mock import patch
-from llamascript import llama
+from lang import llama
 
 
 class TestLlama(unittest.TestCase):