Add missing class with test.