fix small bug
fix small bug