Fix the missing method problem