Fix argument name for S3 method