Fix undefined method error.