diff --git a/tests/unit/test_prepare_discord_query_engine.py b/tests/unit/test_prepare_discord_query_engine.py
index 72ad1a6..45cc348 100644
--- a/tests/unit/test_prepare_discord_query_engine.py
+++ b/tests/unit/test_prepare_discord_query_engine.py
@@ -1,7 +1,7 @@
 import os
 import unittest
 
-from llama_index.core import BaseQueryEngine
+from llama_index.core.base_query_engine import BaseQueryEngine
 from llama_index.vector_stores import ExactMatchFilter, FilterCondition, MetadataFilters
 from utils.query_engine.discord_query_engine import prepare_discord_engine