Added tolerance in the prediction tests