bug fix, add count

bugg fix for add count