diff --git a/tests/test_cli.py b/tests/test_cli.py
index d7f2c93302254167ce68da830c755c05caf54e40..5799e332172e32d27d03ff0e4f3efdc28135e421 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -339,7 +339,7 @@ def test_predict_pasa_montgomery(temporary_basedir, datadir):
             r"^Running prediction on `train` split...$": 1,
             r"^Running prediction on `validation` split...$": 1,
             r"^Running prediction on `test` split...$": 1,
-            r"^Predictions saved to .*$": 1
+            r"^Predictions saved to .*$": 1,
         }
 
         buf.seek(0)
@@ -370,7 +370,7 @@ def test_evaluate_pasa_montgomery(temporary_basedir):
                 "montgomery",
                 f"--predictions={prediction_folder}",
                 f"--output-folder={output_folder}",
-                "--threshold=test"
+                "--threshold=test",
             ],
         )
         _assert_exit_0(result)
@@ -384,7 +384,7 @@ def test_evaluate_pasa_montgomery(temporary_basedir):
             r"^Analyzing split `validation`...$": 1,
             r"^Analyzing split `test`...$": 1,
             r"^Saving measures at .*$": 1,
-            r"^Saving figures at .*$": 1
+            r"^Saving figures at .*$": 1,
         }
         buf.seek(0)
         logging_output = buf.read()