fix test for count