Fix broken test to account of cli changes