diff --git a/tests/model-inversion/nk-2/invert.mod b/tests/model-inversion/nk-2/invert.mod
index 37bf564a4eed5d8db41b531a5dc841ab446ca103..2c19df3af82886eea8b5e642c64cd765943a37a1 100644
--- a/tests/model-inversion/nk-2/invert.mod
+++ b/tests/model-inversion/nk-2/invert.mod
@@ -97,22 +97,22 @@ plot(id, zeros(1,length(id)), 'or')
 hold off
 
 figure(4)
-plot(endogenousvariables.i,'-k','linewidth',2)
+plot(endogenousvariables(1Y:100Y).i,'-k','linewidth',2)
 hold on
-plot(SimulatedData.i(1Y:100Y),'--r','linewidth',2)
+plot(SimulatedData(1Y:100Y).i,'--r','linewidth',2)
 hold off
 title('Nominal interest rate')
 
 figure(5)
-plot(endogenousvariables.y,'-k','linewidth',2)
+plot(endogenousvariables(1Y:100Y).y,'-k','linewidth',2)
 hold on
-plot(SimulatedData.y(1Y:100Y),'--r','linewidth',2)
+plot(SimulatedData(1Y:100Y).y,'--r','linewidth',2)
 hold off
 title('Output gap')
 
 figure(6)
-plot(endogenousvariables.pi,'-k','linewidth',2)
+plot(endogenousvariables(1Y:100Y).pi,'-k','linewidth',2)
 hold on
-plot(SimulatedData.pi(1Y:100Y),'--r','linewidth',2)
+plot(SimulatedData(1Y:100Y).pi,'--r','linewidth',2)
 hold off
 title('Inflation gap')