diff --git a/chunk_codegen.py b/chunk_codegen.py
index 5e2130ee7..77c28fd32 100644
--- a/chunk_codegen.py
+++ b/chunk_codegen.py
@@ -574,7 +574,7 @@ class IndexTracer(object):
             node_idx (int)
         """
         self._assign_index_as_input(node, idx)
-        self._mark_computation(node, idx, [-1, -2])
+        self._mark_computation(node, idx, [-1])
 
     def _assign_elementwise_index(self, node, idx):
         """