diff --git a/bec_widgets/widgets/plots/waveform/waveform.py b/bec_widgets/widgets/plots/waveform/waveform.py
index c2b09864..e50e3a69 100644
--- a/bec_widgets/widgets/plots/waveform/waveform.py
+++ b/bec_widgets/widgets/plots/waveform/waveform.py
@@ -1551,7 +1551,7 @@ class Waveform(PlotBase):
 
         # Iterate over all curves
         for curve in self.curves:
-            if curve.config.source == "custom":
+            if curve.config.source != "device":
                 continue
             dev_name = curve.config.signal.name
             if dev_name in readout_priority_async: