Caio Cesar Daumann
--- a/higgs_dna/systematics/electron_systematics.py 0 → 100644
+++ b/higgs_dna/systematics/electron_systematics.py 0 → 100644
+
+    if is_correction:
+        # scale is a residual correction on data to match MC calibration. Check if is MC, throw error in this case.
+        if hasattr(events, "GenPart"):
+            raise ValueError("Scale corrections should only be applied to data!")
+
+        if year in ["2016preVFP", "2016postVFP", "2017", "2018"]:
+            # the correction is already applied for Run 2
+            logger.info("the scale correction for Run 2  MC is already applied in nAOD, nothing to be done")
+        else:
+            correction = evaluator.evaluate("total_correction", gain, run, eta, r9, ecalEnergy_)
+            ecal_energy_corr = ecalEnergy_ * correction
+
+            corrected_electrons = deepcopy(events.Electron)
+            ecal_energy_corr = ak.unflatten(ecal_energy_corr, counts)
+            corrected_electrons["ele_ecalEnergy"] = ecal_energy_corr