- fix a minor bug