fix undefined method