updated UT depending on code review changes