fixed the solution