569: Improve UT