change return statement