fix a liitle bug