fix: add missing class