diff --git a/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_all.py b/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_all.py
index 49a491995..bbdbc64e7 100644
--- a/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_all.py
+++ b/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_all.py
@@ -28,7 +28,7 @@ async def main():
             predictions.append(str(""))
     df = pd.DataFrame({"Label": labels, "Prediction": predictions, "T/F": is_true})
     df.to_excel("DABench_output.xlsx", index=False)
-    print(DA.eval_all(id_list, predictions))
+    logger.info(DA.eval_all(id_list, predictions))
 
 
 if __name__ == "__main__":
diff --git a/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_sigle.py b/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_sigle.py
index 453c57915..22c0a3f45 100644
--- a/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_sigle.py
+++ b/examples/di/InfiAgent-DABench/run_InfiAgent-DABench_sigle.py
@@ -15,7 +15,7 @@ async def main(id=0):
     logger.info(result)
     save_history(role=di)
     _, is_correct = DA.eval(id, str(result))
-    print(f"Prediction is {'correct' if is_correct else 'incorrect'}.")
+    logger.info(f"Prediction is {'correct' if is_correct else 'incorrect'}.")
 
 
 if __name__ == "__main__":