fix minor bug
fix minor bug