add missing class. Fix #493