fix a bug..