fix small bug;
fix bug;