Minor updates to template to suit test