fixed a small bug.