Ensure we return the accumulator correctly

I moved the INIT check inside to avoid duplication.