fix(benchmark): update benchmark to work with new context.