update tests to work with new CLI