fixed a small bug