fix CLI UT