Fix to calling undefined method.