add missing class reference