diff --git a/src/turbomind/kernels/decoder_multihead_attention/test_decoder_multihead_attention.cu b/src/turbomind/kernels/decoder_multihead_attention/test_decoder_multihead_attention.cu
index f93f8ce466..14c64dba57 100644
--- a/src/turbomind/kernels/decoder_multihead_attention/test_decoder_multihead_attention.cu
+++ b/src/turbomind/kernels/decoder_multihead_attention/test_decoder_multihead_attention.cu
@@ -8,9 +8,9 @@
 #include <iostream>
 #include <thrust/universal_vector.h>
 
+#include <algorithm>
 #include <iomanip>
 #include <numeric>
-#include <random>
 
 using namespace turbomind;