diff --git a/crates/meilisearch/tests/auth/authorization.rs b/crates/meilisearch/tests/auth/authorization.rs
index bc797243b..1e8838e58 100644
--- a/crates/meilisearch/tests/auth/authorization.rs
+++ b/crates/meilisearch/tests/auth/authorization.rs
@@ -294,7 +294,7 @@ async fn unauthorized_partial_actions() {
     let (response, code) = index.create(Some("id")).await;
     assert_eq!(202, code, "{:?}", &response);
     let task_id = response["taskUid"].as_u64().unwrap();
-    index.wait_task(task_id).await.succeeded();
+    server.wait_task(task_id).await.succeeded();
 
     // When multiple actions are necessary, the server mustn't accept any combination with one action missing.
     for ((method, route), actions) in AUTHORIZATIONS.iter() {
diff --git a/crates/meilisearch/tests/documents/render_documents.rs b/crates/meilisearch/tests/documents/render_documents.rs
index c26dc9b64..2aa412adc 100644
--- a/crates/meilisearch/tests/documents/render_documents.rs
+++ b/crates/meilisearch/tests/documents/render_documents.rs
@@ -617,7 +617,7 @@ async fn embedder_document_template() {
     ]);
     let (value, code) = index.add_documents(documents, None).await;
     snapshot!(code, @"202 Accepted");
-    index.wait_task(value.uid()).await.succeeded();
+    server.wait_task(value.uid()).await.succeeded();
 
     let (value, code) = index
         .render(json! {{