fix: undefined method error