partial UT fixes