fixed missing class