fixed tests for count