diff --git a/thirdparty/DiskANN/src/partition_and_pq.cpp b/thirdparty/DiskANN/src/partition_and_pq.cpp
index b135562fc..e3eb0da34 100644
--- a/thirdparty/DiskANN/src/partition_and_pq.cpp
+++ b/thirdparty/DiskANN/src/partition_and_pq.cpp
@@ -532,6 +532,7 @@ int generate_pq_data_from_pivots(const std::string data_file,
             }
           }));
     }
+    knowhere::WaitAllSuccess(futures);
 
     futures.clear();
     futures.reserve(num_pq_chunks);