Added a missing class in templates