Bug Fix:fix a small bug