diff --git a/autochat/test_utils.py b/autochat/test_utils.py
index 85a79d6..23c6a68 100644
--- a/autochat/test_utils.py
+++ b/autochat/test_utils.py
@@ -1,6 +1,6 @@
 import unittest
 
-from utils import inspect_schema, limit_data_size, parse_function
+from autochat.utils import inspect_schema, limit_data_size, parse_function
 
 
 class TestLimitDataSizeUpdated(unittest.TestCase):