Use the appropriate class for validation