Fix the count() test