diff --git a/tests/compare_results.py b/tests/compare_results.py
index ecedb7101515025719231b4445fdd58ae707af17..1c3fe4ea951aef122728a7aed7fc4ecaf8e7607e 100644
--- a/tests/compare_results.py
+++ b/tests/compare_results.py
@@ -8,7 +8,7 @@ import glob
 
 def init_args():
     parser = argparse.ArgumentParser()
-    # params for prediction engine
+    # params for testing assert allclose
     parser.add_argument("--atol", type=float, default=1e-3)
     parser.add_argument("--rtol", type=float, default=1e-3)
     parser.add_argument("--gt_file", type=str, default="")