fix a bug

fix a bug.