diff --git a/tinygrad/engine/jit2.py b/tinygrad/engine/jit2.py
index e77264419a..3665dedfe9 100644
--- a/tinygrad/engine/jit2.py
+++ b/tinygrad/engine/jit2.py
@@ -30,6 +30,7 @@ class TinyJit(Generic[ReturnType]):
     # process schedule caches to be symbolic
     for k,v in input_state_dict.items(): print(k, v.uop.base)
     for input_buffers, sched_cache_key in self.schedule_caches:
+      pre_schedule, combined_sink = schedule_cache[sched_cache_key]
       for k,v in input_buffers.items():
         print(k.pyrender())
         print(v.pyrender())
diff --git a/tinygrad/engine/schedule.py b/tinygrad/engine/schedule.py
index 124323c529..628415fb12 100644
--- a/tinygrad/engine/schedule.py
+++ b/tinygrad/engine/schedule.py
@@ -89,7 +89,7 @@ def create_schedule(sched_sink:UOp) -> tuple[list[ScheduleItem], UOp]:
       else:
         ast, buf_uops, metadata, fixedvars, bound_ranges = si
         fixedvars = fixedvars | {s.src[0].arg[0]:in_ranges[s.src[1]] for s in bound_ranges}
-        pre_schedule.append(ScheduleItem(ast, (), metadata, fixedvars))
+        pre_schedule.append(ScheduleItem(ast, buf_uops, metadata, fixedvars))
         buf_uops_list.append(UOp.sink(*buf_uops))
       sched_ptr += 1
   return pre_schedule, UOp.sink(*buf_uops_list)