fix error in solution