fix a bug in last version