fix a minor bug