diff --git a/tests/test_run.py b/tests/test_run.py
index d0ea6d59880e5da982a949b80818215af5358160..8505b15bb9ba12467282c72eef6c31e112d25545 100644
--- a/tests/test_run.py
+++ b/tests/test_run.py
@@ -64,8 +64,8 @@ expected_scores = {
 @pytest.mark.parametrize(
     "test_input, expected",
     [
-        ((FAKE_ANNOT_BIO, FAKE_PREDICT_BIO, THRESHOLD), expected_scores),
-        ((FAKE_BIO_NESTED, FAKE_BIO_NESTED, THRESHOLD), expected_scores_nested),
+        ((FAKE_ANNOT_BIO, FAKE_PREDICT_BIO, THRESHOLD, True), expected_scores),
+        ((FAKE_BIO_NESTED, FAKE_BIO_NESTED, THRESHOLD, True), expected_scores_nested),
     ],
 )
 def test_run(test_input, expected):