Update tests after template changes