[UPDATE] fixed missing class