fix bug

fix a bug