diff --git a/tests/test_token_counter.py b/tests/test_token_counter.py
index d13f2ae0..42670e60 100644
--- a/tests/test_token_counter.py
+++ b/tests/test_token_counter.py
@@ -1,7 +1,7 @@
 import unittest
 import tests.context
+from autogpt.token_counter import count_message_tokens, count_string_tokens
 
-from scripts.token_counter import count_message_tokens, count_string_tokens
 
 class TestTokenCounter(unittest.TestCase):