small bug fxes