Added missing template include