diff --git a/lib/src/main/groovy/com/jessebrault/ssg/template/TemplateFileTemplatesProvider.groovy b/lib/src/main/groovy/com/jessebrault/ssg/template/TemplateFileTemplatesProvider.groovy index 2bfae98..bbcdcea 100644 --- a/lib/src/main/groovy/com/jessebrault/ssg/template/TemplateFileTemplatesProvider.groovy +++ b/lib/src/main/groovy/com/jessebrault/ssg/template/TemplateFileTemplatesProvider.groovy @@ -13,11 +13,11 @@ class TemplateFileTemplatesProvider implements TemplatesProvider { private static final Logger logger = LoggerFactory.getLogger(TemplateFileTemplatesProvider) - private final Collection types + private final Collection templateTypes private final File templatesDir private TemplateType getType(File file) { - this.types.find { + this.templateTypes.find { it.extensions.contains(new FileNameHandler(file).getExtension()) } }