Redefine schedule_map to apply a schedule per-update