Fix missing class...?