#578: improve UT