fixed spec to return count