modify a small bug