- fix minor bug