diff --git a/quam/components/pulses.py b/quam/components/pulses.py
index 1987bc3..d6f9609 100644
--- a/quam/components/pulses.py
+++ b/quam/components/pulses.py
@@ -12,6 +12,7 @@
     "Pulse",
     "BaseReadoutPulse",
     "ReadoutPulse",
+    "WaveformPulse",
     "DragGaussianPulse",
     "DragCosinePulse",
     "DragPulse",