unittest fixes