diff --git a/cpp/bench/neighbors/knn.cuh b/cpp/bench/neighbors/knn.cuh
index 0a59da3970..259d39d8f7 100644
--- a/cpp/bench/neighbors/knn.cuh
+++ b/cpp/bench/neighbors/knn.cuh
@@ -26,13 +26,10 @@
 
 #if defined RAFT_DISTANCE_COMPILED
 #include <raft/neighbors/specializations.cuh>
-
-// TODO: Legacy. Remove when FAISS is removed
-#include <raft/spatial/knn/specializations.cuh>
 #endif
 
 #if defined RAFT_NN_COMPILED
-#include <raft/neighbors/specializations.cuh>
+// TODO: Legacy. Remove when FAISS is removed
 #include <raft/spatial/knn/specializations.cuh>
 #endif