fixed ssmall bug