diff --git a/tests/unit/test_website_query_engine.py b/tests/unit/test_website_query_engine.py
index 056d76d..17d5b18 100644
--- a/tests/unit/test_website_query_engine.py
+++ b/tests/unit/test_website_query_engine.py
@@ -9,9 +9,9 @@
 class TestNotionQueryEngine(TestCase):
     def setUp(self) -> None:
         community_id = "sample_community"
-        self.notion_query_engine = WebsiteQueryEngine(community_id)
+        self.website_query_engine = WebsiteQueryEngine(community_id)
 
     def test_prepare_engine(self):
-        notion_query_engine = self.notion_query_engine.prepare(testing=True)
-        print(notion_query_engine.__dict__)
-        self.assertIsInstance(notion_query_engine.retriever, VectorIndexRetriever)
+        website_query_engine = self.website_query_engine.prepare(testing=True)
+        print(website_query_engine.__dict__)
+        self.assertIsInstance(website_query_engine.retriever, VectorIndexRetriever)