Fix loggregator test to reflect new CLI changes