minor UT fix