fixes

small bug fixes