Fix bug where we didn't wait for benchmark to finish after warmup