add missing class use