Add missing class, fixing #25