fix error bug