fix a bug...