diff --git a/src/mlpack/methods/random_forest/random_forest_main.cpp b/src/mlpack/methods/random_forest/random_forest_main.cpp
index f64fdc8c07..c73385a709 100644
--- a/src/mlpack/methods/random_forest/random_forest_main.cpp
+++ b/src/mlpack/methods/random_forest/random_forest_main.cpp
@@ -207,9 +207,9 @@ static void mlpackMain()
   ReportIgnoredParam({{ "training", false }}, "minimum_leaf_size");
 
   RandomForestModel* rfModel;
-  // Handles the case when we are either training on top of existing forest
-  // else we are making predictions only.
-  if (IO::HasParam("warm_start") or IO::HasParam("input_model"))
+  // Input model is loaded when we are either doing warm-started training or
+  // else we are making predictions only or both.
+  if (IO::HasParam("input_model"))
     rfModel = IO::GetParam<RandomForestModel*>("input_model");
   // Handles the case when we are training new forest from scratch.
   else