added code to fix error