Edit wrong math function