diff --git a/rdagent/components/coder/data_science/ensemble/eval_tests/ensemble_test.txt b/rdagent/components/coder/data_science/ensemble/eval_tests/ensemble_test.txt
index b9b0042f..c0f624ae 100644
--- a/rdagent/components/coder/data_science/ensemble/eval_tests/ensemble_test.txt
+++ b/rdagent/components/coder/data_science/ensemble/eval_tests/ensemble_test.txt
@@ -77,5 +77,5 @@ for model in {{model_names}}:
 print("Please check scores dataframe's format, it's:")
 print(score_df)
 
-print("Ensemble test passed successfully.")
-print(f"Output shape: {final_pred.shape}")
+print("Ensemble test end.")
+print(f"Final prediction shape: {final_pred.shape}")
diff --git a/rdagent/components/coder/data_science/model/eval_tests/model_test.txt b/rdagent/components/coder/data_science/model/eval_tests/model_test.txt
index 9ddcd183..2a1f8a7f 100644
--- a/rdagent/components/coder/data_science/model/eval_tests/model_test.txt
+++ b/rdagent/components/coder/data_science/model/eval_tests/model_test.txt
@@ -50,4 +50,4 @@ val_pred, test_pred, final_hypers = model_workflow(
 )
 log_execution_results(start_time, val_pred, test_pred, final_hypers, "The second execution")
 
-print("Model code test passed successfully.")
+print("Model code test end.")