correct a small bug