diff --git a/benchmarks/clustering/scripts/gen_particles.py b/benchmarks/clustering/scripts/gen_particles.py
index 2d0b81c8278d345cff479161f107f08f62c55bf1..620bf0aafff171ab9c797192ee8df386960e610f 100644
--- a/benchmarks/clustering/scripts/gen_particles.py
+++ b/benchmarks/clustering/scripts/gen_particles.py
@@ -1,4 +1,5 @@
 import os
+import sys
 from pyHepMC3 import HepMC3 as hm
 import numpy as np
 import argparse
diff --git a/benchmarks/ecal/scripts/gen_particles.py b/benchmarks/ecal/scripts/gen_particles.py
index 2d0b81c8278d345cff479161f107f08f62c55bf1..620bf0aafff171ab9c797192ee8df386960e610f 100644
--- a/benchmarks/ecal/scripts/gen_particles.py
+++ b/benchmarks/ecal/scripts/gen_particles.py
@@ -1,4 +1,5 @@
 import os
+import sys
 from pyHepMC3 import HepMC3 as hm
 import numpy as np
 import argparse
diff --git a/benchmarks/imaging_ecal/scripts/gen_particles.py b/benchmarks/imaging_ecal/scripts/gen_particles.py
index fb3a8966384f597e258094b342be738e32805e4f..a357b752ca227ece1103c193be3b334eb40f2e26 100644
--- a/benchmarks/imaging_ecal/scripts/gen_particles.py
+++ b/benchmarks/imaging_ecal/scripts/gen_particles.py
@@ -1,4 +1,5 @@
 import os
+import sys
 from pyHepMC3 import HepMC3 as hm
 import numpy as np
 import argparse
diff --git a/benchmarks/imaging_shower_ML/scripts/gen_particles.py b/benchmarks/imaging_shower_ML/scripts/gen_particles.py
index af22f35d6e642ca5b1e8e809d6bf32f7d2eca1de..372b2c829d61c4a35ce58810d43705f289bc7fd0 100644
--- a/benchmarks/imaging_shower_ML/scripts/gen_particles.py
+++ b/benchmarks/imaging_shower_ML/scripts/gen_particles.py
@@ -1,4 +1,5 @@
 import os
+import sys
 from pyHepMC3 import HepMC3 as hm
 import numpy as np
 import argparse
diff --git a/benchmarks/tracking/scripts/gen_particles.py b/benchmarks/tracking/scripts/gen_particles.py
index 4b8c4e96b1dcbfa81e11c740f41854b81c0ad6ee..3b86efca20e3a6060cbd1848149c92f59ead5e3a 100644
--- a/benchmarks/tracking/scripts/gen_particles.py
+++ b/benchmarks/tracking/scripts/gen_particles.py
@@ -1,4 +1,5 @@
 import os
+import sys
 from pyHepMC3 import HepMC3 as hm
 import numpy as np
 import argparse