fix one small bug