Update test template to take args