diff --git a/pylot/core/util/widgets.py b/pylot/core/util/widgets.py
index 0139e457..63458d80 100644
--- a/pylot/core/util/widgets.py
+++ b/pylot/core/util/widgets.py
@@ -779,7 +779,7 @@ class PickDlg(QDialog):
                 trace = selectTrace(trace, '12')
                 if trace:
                     wfdata.append(trace)
-        elif component == 'Z':
+        else:
             wfdata = self.getWFData().select(component=component)
         return wfdata