fix small bug