diff --git a/tests/test_ivf_index.cpp b/tests/test_ivf_index.cpp
index 54cb7945f9..8e4530a956 100644
--- a/tests/test_ivf_index.cpp
+++ b/tests/test_ivf_index.cpp
@@ -12,6 +12,8 @@
 #include <cstring>
 #include <map>
 #include <random>
+#include <set>
+#include <algorithm>
 
 #include <gtest/gtest.h>