Updated include to avoid duplicate declaration of class