minor fix for the UT