Unittest fixes