--- /var/folders/n8/bv7zgpwx3b9crc3mbp6zqb5w0000gn/T/tmp.eWO02SiwSF/coremltools/converters/mil/mil/passes/defs/optimize_state.py	2025-11-03 23:10:10
+++ /private/tmp/claude-501/-Users-aiden-Developer-Kino-kino/54f28767-a77e-430f-881b-85e18da4ab2e/scratchpad/.venv/lib/python3.9/site-packages/coremltools/converters/mil/mil/passes/defs/optimize_state.py	2026-07-23 21:53:13
@@ -87,7 +87,18 @@
 
         coreml_update_state_ops = self._try_find_child_coreml_update_state_ops(op)
         for coreml_update_state_op in coreml_update_state_ops:
-            before_op = block_operation_list[block_operation_list.index(op) + 1]
+            # PATCued: block_operation_list is a stale snapshot; earlier iterations
+            # may have removed ops (their enclosing_block is None). Walk forward to
+            # the next op still present in the block, else insert at block end.
+            idx = block_operation_list.index(op) + 1
+            while (
+                idx < len(block_operation_list)
+                and block_operation_list[idx].enclosing_block is None
+            ):
+                idx += 1
+            before_op = (
+                block_operation_list[idx] if idx < len(block_operation_list) else None
+            )
             scopes = self._construct_scope_info_list_from_op_scopes(op)
             with mb.scope(*scopes):
                 immediate_coreml_update_state = mb.coreml_update_state(