fixing broken UT