fixed the return problem