diff --git a/cognee/tests/integration/retrieval/test_completion.py b/cognee/tests/unit/modules/retrieval/test_completion.py
similarity index 100%
rename from cognee/tests/integration/retrieval/test_completion.py
rename to cognee/tests/unit/modules/retrieval/test_completion.py
diff --git a/cognee/tests/integration/retrieval/test_graph_summary_completion_retriever.py b/cognee/tests/unit/modules/retrieval/test_graph_summary_completion_retriever.py
similarity index 100%
rename from cognee/tests/integration/retrieval/test_graph_summary_completion_retriever.py
rename to cognee/tests/unit/modules/retrieval/test_graph_summary_completion_retriever.py
diff --git a/cognee/tests/integration/retrieval/test_user_qa_feedback.py b/cognee/tests/unit/modules/retrieval/test_user_qa_feedback.py
similarity index 100%
rename from cognee/tests/integration/retrieval/test_user_qa_feedback.py
rename to cognee/tests/unit/modules/retrieval/test_user_qa_feedback.py