add missing class from previous commit