Fix error in build
Cause by missing class