fix error for problem4