fix a liitle bug.