corrected small bug