diff --git a/libs/neo4j/tests/unit_tests/llms/__init__.py b/libs/neo4j/tests/integration_tests/chains/__init__.py
similarity index 100%
rename from libs/neo4j/tests/unit_tests/llms/__init__.py
rename to libs/neo4j/tests/integration_tests/chains/__init__.py
diff --git a/libs/neo4j/tests/integration_tests/chains/test_graph_database.py b/libs/neo4j/tests/integration_tests/chains/test_graph_database.py
index 1f4046a..eda5683 100644
--- a/libs/neo4j/tests/integration_tests/chains/test_graph_database.py
+++ b/libs/neo4j/tests/integration_tests/chains/test_graph_database.py
@@ -7,7 +7,7 @@
 
 from langchain_neo4j.chains.graph_qa.cypher import GraphCypherQAChain
 from langchain_neo4j.graphs.neo4j_graph import Neo4jGraph
-from tests.unit_tests.llms.fake_llm import FakeLLM
+from tests.llms.fake_llm import FakeLLM
 
 
 def test_connect_neo4j() -> None:
diff --git a/libs/neo4j/tests/llms/__init__.py b/libs/neo4j/tests/llms/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/libs/neo4j/tests/unit_tests/llms/fake_llm.py b/libs/neo4j/tests/llms/fake_llm.py
similarity index 100%
rename from libs/neo4j/tests/unit_tests/llms/fake_llm.py
rename to libs/neo4j/tests/llms/fake_llm.py
diff --git a/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py b/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py
index 030d5cf..34d1f38 100644
--- a/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py
+++ b/libs/neo4j/tests/unit_tests/chains/test_graph_qa.py
@@ -34,7 +34,7 @@
 )
 from langchain_neo4j.graphs.graph_document import GraphDocument
 from langchain_neo4j.graphs.graph_store import GraphStore
-from tests.unit_tests.llms.fake_llm import FakeLLM
+from tests.llms.fake_llm import FakeLLM
 
 
 class FakeGraphStore(GraphStore):