Fix call to undefined method