PredictSlice was not rescaling outputs