Fix the waffle badge to show the correct count