small bug corrected