diff --git a/modules/benchmarks_external/src/tree/tree_benchmark.es6 b/modules/benchmarks_external/src/tree/tree_benchmark.es6
index c30b3f2286..641dcd35ac 100644
--- a/modules/benchmarks_external/src/tree/tree_benchmark.es6
+++ b/modules/benchmarks_external/src/tree/tree_benchmark.es6
@@ -2,7 +2,7 @@
 import {getIntParameter, bindAction} from 'e2e_test_lib/benchmark_util';
 
 export function main() {
-  angular.bootstrap(document.body, ['app']);
+  angular.bootstrap(document.querySelector('tree'), ['app']);
 }
 
 angular.module('app', [])