Updating @count to count() in tests