diff --git a/src/IQC.jl b/src/IQC.jl
index cbd690e5c04e194afe51ad62be3000e4e535b265..54f8b5493c97ac0afc9dab8b347bb72e392b168d 100644
--- a/src/IQC.jl
+++ b/src/IQC.jl
@@ -5224,13 +5224,13 @@ end
 
 
 # %%
-function gradient(sys :: DynamicSystem, k; margin = 1e-3)
+function gradient(sys :: DynamicSystem, k; λ = 0.0, margin = 1e-3)
     eigen = LinearAlgebra.eigen
     I = LinearAlgebra.I
     
     (; A, B, C, Pi) = generate_feasibility_problem_for_KYP(sys)
 
-    (success, P) = solve!(sys, margin = margin, req_PSD_P=true, silent=true)
+    (success, P) = solve!(sys, λ = λ, margin = margin, req_PSD_P=true, silent=true)
     
     lhs = [ A'*P+P*A P*B; B'*P 0.0*I ] + Pi + margin*I