fixed a minor bug...