fixed a minor bug