Fixed missing method error