Minor updates to validate_args decorator