fixed missing class definition