diff --git a/exir/tests/test_memory_planning.py b/exir/tests/test_memory_planning.py
index 1e9a7e29f30..d5e5627dfa6 100644
--- a/exir/tests/test_memory_planning.py
+++ b/exir/tests/test_memory_planning.py
@@ -756,18 +756,18 @@ def forward(self, input, label):
 
         ep.dump_executorch_program(True)
 
-        # 155 just so happens to be the index of the user_grad output arg of
+        # 147 just so happens to be the index of the user_grad output arg of
         # convolution_backward.out. This is fairly fragile.
         # Check that the None output is not memory planned.
         self.assertEqual(
             ep.executorch_program.execution_plan[0]
-            .values[155]
+            .values[147]
             .val.data_buffer_idx,  # pyright: ignore
             0,
         )
         self.assertEqual(
             ep.executorch_program.execution_plan[0]
-            .values[155]
+            .values[147]
             .val.allocation_info,  # pyright: ignore
             None,
         )