Remove count assertions