diff --git a/src/calng/correction_addons/peakfinder9.py b/src/calng/correction_addons/peakfinder9.py
index b4694ce21df9eb72d41a16f4a7b78e487e799733..dba7421411a5b64371ffef2bbc99aaf9836685ae 100644
--- a/src/calng/correction_addons/peakfinder9.py
+++ b/src/calng/correction_addons/peakfinder9.py
@@ -124,7 +124,7 @@ class Peakfinder9(BaseCorrectionAddon):
             .commit(),
         )
 
-    def post_correction(self, data, train_id, cell_table, pulse_table, output_hash):
+    def post_correction(self, train_id, data, cell_table, pulse_table, output_hash):
         # assumes processed data shape is frames, pixels, pixels
         if self._input_shape != data.shape:
             try: