diff --git a/src/include/alloc.h b/src/include/alloc.h
index 8456c4bd69..6ebe828a36 100644
--- a/src/include/alloc.h
+++ b/src/include/alloc.h
@@ -17,7 +17,7 @@
 #include <unistd.h>
 #include <stdlib.h>
 #include <string.h>
-#include <map>
+#include <unordered_map>
 #include "rccl_vars.h"
 
 #if CUDART_VERSION >= 11030