diff --git a/deeplake/core/vectorstore/test_deeplake_vectorstore.py b/deeplake/core/vectorstore/test_deeplake_vectorstore.py
index bc5a28b726..a4a5c1cd4a 100644
--- a/deeplake/core/vectorstore/test_deeplake_vectorstore.py
+++ b/deeplake/core/vectorstore/test_deeplake_vectorstore.py
@@ -1754,7 +1754,7 @@ def test_query_dim(local_path):
     )
 
     vector_store.add(text=texts, embedding=embeddings)
-    with pytest.raises(AssertionError):
+    with pytest.raises(NotImplementedError):
         vector_store.search([texts[0], texts[0]], embedding_fn3, k=1)
 
     vector_store.search([texts[0]], embedding_fn4, k=1)