diff --git a/tests/e2e/conftest.py b/tests/e2e/conftest.py
index 442a51a02..3176003a1 100644
--- a/tests/e2e/conftest.py
+++ b/tests/e2e/conftest.py
@@ -56,7 +56,7 @@ def setup_neo4j(driver):
         vector_index_name,
         label="Document",
         property="propertyKey",
-        dimensions=10,
+        dimensions=1536,
         similarity_fn="euclidean",
     )
 
@@ -66,7 +66,7 @@ def setup_neo4j(driver):
     )
 
     # Insert 10 vectors and authors
-    vector = [random.random() for _ in range(10)]
+    vector = [random.random() for _ in range(1536)]
 
     def random_str(n: int) -> str:
         return "".join([random.choice(string.ascii_letters) for _ in range(n)])
@@ -88,6 +88,6 @@ def random_str(n: int) -> str:
             "id": str(uuid.uuid4()),
             "i": i,
             "vector": vector,
-            "authorName": random_str(10),
+            "authorName": random_str(1536),
         }
         driver.execute_query(insert_query, parameters)
diff --git a/tests/e2e/test_vector_e2e.py b/tests/e2e/test_vector_e2e.py
index baeae1915..608dd4d03 100644
--- a/tests/e2e/test_vector_e2e.py
+++ b/tests/e2e/test_vector_e2e.py
@@ -113,7 +113,7 @@ def test_vector_retriever_filters(driver):
 
     top_k = 2
     results = retriever.search(
-        query_vector=[1.0 for _ in range(10)],
+        query_vector=[1.0 for _ in range(1536)],
         filters={"int_property": {"$gt": 2}},
         top_k=top_k,
     )