return problem fixed