[cli] Fix broken tests.