fix(profile): use context.state

unjs · Nov 9, 2018 · 0d92aae · 0d92aae
1 parent 4f28403
commit 0d92aae
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/reporters/profile.js b/src/reporters/profile.js
@@ -5,17 +5,17 @@ import Profiler from '../profiler';
 
 export default class ProfileReporter {
   progress(context) {
-    if (!context.profiler) {
-      context.profiler = new Profiler();
+    if (!context.state.profiler) {
+      context.state.profiler = new Profiler();
     }
 
-    context.profiler.onRequest(context.state.request);
+    context.state.profiler.onRequest(context.state.request);
   }
 
   done(context) {
-    if (context.profiler) {
-      context.state.profile = context.profiler.getFormattedStats();
-      delete context.profiler;
+    if (context.state.profiler) {
+      context.state.profile = context.state.profiler.getFormattedStats();
+      delete context.state.profiler;
     }
   }