Fix calling undefined method