ehh... add missing class