diff --git a/tests/hipify-clang/axpy.cu b/tests/hipify-clang/axpy.cu
index 4dfabff7e7..071c503a35 100644
--- a/tests/hipify-clang/axpy.cu
+++ b/tests/hipify-clang/axpy.cu
@@ -2,6 +2,8 @@
 
 #include <iostream>
 
+// CHECK: #include <hip/hip_runtime.h>
+#include <cuda.h>
 
 #define TOKEN_PASTE(X, Y) X ## Y
 #define ARG_LIST_AS_MACRO a, device_x, device_y