This should do it...
Fix missing class