Minor amend to clarify param type