Small fix for missing class.