Add missing count increment