small bug corrections