Modified the hardcoded TEMPLATE_DIRS