add missing class from previous source code