fix include for new class name