Update tests for new command line args.