Fix comment count update