Remove undefined method error