Update CLI tests