fixed missing class reference

Signed-off-by: Raf <7909000@gmail.com>