diff --git a/src/cal_tools/agipdlib.py b/src/cal_tools/agipdlib.py
index bacf400ba985b8672295e14ebb97887b2f5d8cc8..d2a6bbed351fd6c56afb673a01998dc1c070b4ba 100644
--- a/src/cal_tools/agipdlib.py
+++ b/src/cal_tools/agipdlib.py
@@ -1416,10 +1416,9 @@ class AgipdCorrections:
             return
 
     def initialize_from_db(self, karabo_id: str, karabo_da: str,
-                           cal_db_interface: str,
-                           creation_time: datetime,
-                           memory_cells: float, bias_voltage: int,
-                           gain_setting: float, acquisition_rate: float,
+                           creation_time: datetime, memory_cells: float,
+                           bias_voltage: int, gain_setting: int,
+                           acquisition_rate: float, gain_mode: int,
                            integration_time: int, module_idx: int,
                            photon_energy: float = 9.2, client = None,
     ):
@@ -1477,7 +1476,7 @@ class AgipdCorrections:
             acquisition_rate=acquisition_rate,
             integration_time=integration_time,
             source_energy=photon_energy,
-            gain_mode=self.gain_mode,
+            gain_mode=gain_mode,
             gain_setting=gain_setting,
             event_at=creation_time,
             client=client,
@@ -1487,7 +1486,7 @@ class AgipdCorrections:
             if cname in self.gain_constants:
                 agipd_cal.gain_mode = None
             else:
-                agipd_cal.gain_mode = self.gain_mode
+                agipd_cal.gain_mode = gain_mode
             try:
                 agipd_md = agipd_cal.metadata([cname])[karabo_da]
             except CalCatError as e: