Add missing method for fixing compilation error