From c6addc9da60ff9c811191723da06faddb246f4bf Mon Sep 17 00:00:00 2001
From: Johannes Pfeifer <jpfeifer@gmx.de>
Date: Fri, 22 Oct 2021 16:59:25 +0200
Subject: [PATCH] stochastic_solvers.m: fix bug introduced in 57bbec1f

Rely on lead_lag_incidence instead of kstate to identify states

(cherry picked from commit 17a58ccec60774ebf33c8cf3fc1ec4769ecf17e2)
---
 matlab/stochastic_solvers.m | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/matlab/stochastic_solvers.m b/matlab/stochastic_solvers.m
index cd9a839bde..62fb0e7041 100644
--- a/matlab/stochastic_solvers.m
+++ b/matlab/stochastic_solvers.m
@@ -216,7 +216,11 @@ if M_.maximum_endo_lead == 0
     % backward models: simplified code exist only at order == 1
     if local_order == 1
         [k1,~,k2] = find(kstate(:,4));
-        dr.state_var = k2;
+        if M_.maximum_endo_lag
+            dr.state_var = find(M_.lead_lag_incidence(1,:));
+        else
+            dr.state_var = [];
+        end
         dr.ghx(:,k1) = -b\jacobia_(:,k2);
         if M_.exo_nbr
             dr.ghu =  -b\jacobia_(:,nz+1:end);
-- 
GitLab