fix a partial bug.