fix minor bug

fix minor bug