diff --git a/scripts/ztplot.py b/scripts/ztplot.py
index 91112eae2c57070c35a1d4b69d1093b4f32272fd..24aca0f4509423c731502b964334c6b32c8a563a 100755
--- a/scripts/ztplot.py
+++ b/scripts/ztplot.py
@@ -32,6 +32,7 @@ import matplotlib.ticker as ticker
 import numpy as np
 
 import km3pipe as kp
+from km3pipe.io.daq import is_3dmuon, is_3dshower, is_mxshower
 from km3modules.hits import count_multiplicities
 import km3pipe.style
 km3pipe.style.use('km3pipe')