diff --git a/src/gradients/pulsed_gradients.jl b/src/gradients/pulsed_gradients.jl
index bf66128045d70c8e96897e55968dd790ccdd1bc1..afcc1f3d2519cd8956ca606682b2e39eab6e14a5 100644
--- a/src/gradients/pulsed_gradients.jl
+++ b/src/gradients/pulsed_gradients.jl
@@ -91,8 +91,8 @@ function PulsedGradient(model::Model; orientation=nothing, rise_time=nothing, fl
     )
 
     set_simple_constraints!(model, res, kwargs)
-    @constraint model res.flat_time >= 0
-    @constraint model res.rise_time >= 0
+    @constraint model flat_time >= 0
+    @constraint model rise_time >= 0
     return res
 end