fixed a minor bug?