Custom error, add count tests