Fix offset updates for outer_fold