fixing missing class