fix error in code