fixed a small bug.......