diff --git a/tests/unit_tests/parsers/test_fragpipe.py b/tests/unit_tests/parsers/test_fragpipe.py
index 59e5f72..19cf357 100644
--- a/tests/unit_tests/parsers/test_fragpipe.py
+++ b/tests/unit_tests/parsers/test_fragpipe.py
@@ -1,5 +1,3 @@
-# Generation assisted by CodiumAI
-
 from picked_group_fdr.parsers.fragpipe import parse_fragpipe_psm_file
 from picked_group_fdr.scoring_strategy import ProteinScoringStrategy
 
@@ -10,7 +8,10 @@
 class TestParseFragpipePsmFile:
 
     # correctly parses a well-formed psm.tsv file with all required columns
-    def test_correctly_parses_well_formed_psm_tsv(self, mocker):
+    @pytest.mark.parametrize(
+        "probability_column", ["PeptideProphet Probability", "Probability"]
+    )
+    def test_correctly_parses_well_formed_psm_tsv(self, mocker, probability_column):
         # Mock the csv reader
         mocker.patch(
             "picked_group_fdr.parsers.tsv.get_tsv_reader",
@@ -22,7 +23,7 @@ def test_correctly_parses_well_formed_psm_tsv(self, mocker):
                             "Peptide",
                             "Modified Peptide",
                             "SpectralSim",
-                            "PeptideProphet Probability",
+                            probability_column,
                             "Protein",
                             "Mapped Proteins",
                         ],