Switch to annotation based params that was somehow missed.
Merge pull request #411 from william-ferguson-au/fix_for_issue_410

Switch to annotation based params that was somehow missed.