For improved benchmarking accuracy, shouldn't we include this in the args.benchmark branch?