Fix error in build
Cause by missing method