diff --git a/test/test_fori_loop_simple_linear_model_test_code.py b/test/test_fori_loop_simple_linear_model_test_code.py
index 9866fc90526..4c0f96526fe 100644
--- a/test/test_fori_loop_simple_linear_model_test_code.py
+++ b/test/test_fori_loop_simple_linear_model_test_code.py
@@ -40,7 +40,9 @@ def body_fun(l_in_i):
 print("l_in_i_plus_1_: ", l_in_i_plus_1_)
 print("l_out_: ", l_out_)
 
-
+for i in range(5):
+  l_out_expected = linear_0(l_in_0)
+print("expected: ", l_out_expected)
 
 # --- linear two ---
 l_in_2 = torch.randn(10, device=xm.xla_device())