diff --git a/src/objective/aft_obj.cu b/src/objective/aft_obj.cu
index 9c34b827a..795f75bcc 100644
--- a/src/objective/aft_obj.cu
+++ b/src/objective/aft_obj.cu
@@ -28,9 +28,9 @@ using AFTLoss = xgboost::common::AFTLoss<Distribution>;
 namespace xgboost {
 namespace obj {
 
-#if defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
+#if defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
 DMLC_REGISTRY_FILE_TAG(aft_obj_gpu);
-#endif  // defined(XGBOOST_USE_CUDA) || !defined(XGBOOST_USE_HIP)
+#endif  // defined(XGBOOST_USE_CUDA) || defined(XGBOOST_USE_HIP)
 
 class AFTObj : public ObjFunction {
  public: