repaired small bug