Remove return statement