Fix undefined method name error