Minor fix/hack for setting notification count