update
fix small bug