diff --git a/src/mlpack/methods/decision_tree/decision_tree_main.cpp b/src/mlpack/methods/decision_tree/decision_tree_main.cpp
index f8eee66c4d..8369312d1d 100644
--- a/src/mlpack/methods/decision_tree/decision_tree_main.cpp
+++ b/src/mlpack/methods/decision_tree/decision_tree_main.cpp
@@ -181,21 +181,29 @@ static void mlpackMain()
       arma::Row<double> weights =
           std::move(CLI::GetParam<arma::Mat<double>>("weights"));
       if (CLI::HasParam("print_training_error"))
+      {
         model->tree = DecisionTree<>(trainingSet, model->info, labels,
             numClasses, std::move(weights), minLeafSize, minimumGainSplit);
+      }
       else
+      {
         model->tree = DecisionTree<>(std::move(trainingSet), model->info,
             std::move(labels), numClasses, std::move(weights), minLeafSize,
             minimumGainSplit);
+      }
     }
     else
     {
       if (CLI::HasParam("print_training_error"))
+      {
         model->tree = DecisionTree<>(trainingSet, model->info, labels,
             numClasses, minLeafSize, minimumGainSplit);
+      }
       else
+      {
         model->tree = DecisionTree<>(std::move(trainingSet), model->info,
             std::move(labels), numClasses, minLeafSize, minimumGainSplit);
+      }
     }
 
     // Do we need to print training error?