fixed failing UT