corrected a small bug