Updated argument name for new API