diff --git a/tests/cpp/objective/test_aft_obj.cu b/tests/cpp/objective/test_aft_obj.cu
index 3da6bc94b9f7..134d1a6b7e5e 100644
--- a/tests/cpp/objective/test_aft_obj.cu
+++ b/tests/cpp/objective/test_aft_obj.cu
@@ -3,4 +3,4 @@
  */
 // Dummy file to keep the CUDA tests.
 
-#include "test_aft_obj.cc"
+#include "test_aft_obj_cpu.cc"