fix a bug

fix bug