fix small bug .