Fix the template inclusion bug