Bug fix for missing class.