diff --git a/src/nqrduck_spectrometer_limenqr/model.py b/src/nqrduck_spectrometer_limenqr/model.py
index 015e31b..ad38834 100644
--- a/src/nqrduck_spectrometer_limenqr/model.py
+++ b/src/nqrduck_spectrometer_limenqr/model.py
@@ -1,7 +1,7 @@
 import logging
 from nqrduck.module.module_model import ModuleModel
 from nqrduck_spectrometer.base_spectrometer_model import BaseSpectrometerModel
-from nqrduck_spectrometer.pulseparameters import Gate
+from nqrduck_spectrometer.pulseparameters import Gate, TXPulse
 
 logger = logging.getLogger(__name__)
 
@@ -16,8 +16,8 @@ class LimeNQRModel(BaseSpectrometerModel):
         self.add_setting("Dwell time", "400n", "Dwell time")
         self.add_setting("RX Gain", 55, "RX Gain")
         self.add_setting("TX Gain", 40, "TX Gain")
-        self.add_pulse_parameter_option("TX Gate", Gate)
-        self.add_pulse_parameter_option("RX Gate", Gate)
+        self.add_pulse_parameter_option("TX", TXPulse)
+        self.add_pulse_parameter_option("Gate", Gate)
 
         try:
             from nqrduck_pulseprogrammer.pulseprogrammer import pulse_programmer