Use the proper key in the test -- compute() is listed as "initial batch" in the profiler
Enable aggregation of profiler logs