Fixing TemplateUnit to TemplateType

Fixing TemplateUnit to TemplateType