Added count assertion