diff --git a/test/functionality_tests.jl b/test/functionality_tests.jl
index 3c3dca0a..18b1082c 100644
--- a/test/functionality_tests.jl
+++ b/test/functionality_tests.jl
@@ -292,8 +292,8 @@ function functionality_test(m; algorithm = :first_order, plots = true, verbose =
     end
 
     conditions = spzeros(size(new_sub_irfs_all,1),2)
-    conditions[1,1] = .01
-    conditions[2,2] = .02
+    conditions[var_idxs[1],1] = .01
+    conditions[var_idxs[2],2] = .02
 
     cond_fcst = get_conditional_forecast(m, conditions, algorithm = algorithm, conditions_in_levels = false)