fix a small bug