Fix missing class