Fix undefined method