bug fix, add count