add fix small bug