Solved change return problem