Update template to use correct parameter