re-add correct count