couple small bug fixes