diff --git a/scripts/write_parametrized.py b/scripts/write_parametrized.py
index 3fcece74d1fb590451ca81761c6f6f1c18f33aef..a0a0ab8b8805b9f2d99c44edee2c9e5804bb6317 100755
--- a/scripts/write_parametrized.py
+++ b/scripts/write_parametrized.py
@@ -46,6 +46,7 @@ if __name__ == "__main__":
             params = tuple([float(i) for i in match.groups()])
             if not params in count_dict:
                 count_dict[params] = 0
+            # TODO: might be better to use sum of weights
             count_dict[params] += tree.GetEntries()
     f.Close()
 
@@ -71,9 +72,11 @@ if __name__ == "__main__":
                     param_values = numbers[rnd_idx]
                     for param_idx, param_name in enumerate(param_names):
                         df[param_name] = param_values[:,param_idx]
+                    df["training_weight"] = df["eventWeight"]*df["genWeight"]
                 else:
                     for param_name, param_value in zip(param_names, match_signal.groups()):
                         df[param_name] = float(param_value)
+                    df["training_weight"] = df["eventWeight"]
                 if match_signal is None:
                     out_tree_name = tree_name
                 else: