Updated template.
Added missing import into the template.