diff --git a/spec/meilisearch/index/documents_spec.rb b/spec/meilisearch/index/documents_spec.rb
index 177af894..bc558cbd 100644
--- a/spec/meilisearch/index/documents_spec.rb
+++ b/spec/meilisearch/index/documents_spec.rb
@@ -157,7 +157,7 @@
           client.create_index('vector_test').await
           new_index = client.index('vector_test')
           new_index.add_documents(new_doc).await
-          expect(new_index.document(123)['_vectors']).to include(0.1)
+          expect(new_index.get_documents(id: 123, retrieveVector: true)['_vectors']).to include(0.1)
         end
       end
     end