diff --git a/src/tracing/agent.cc b/src/tracing/agent.cc
index 1ac99bbb34bbbf..bc04b0b5e60d17 100644
--- a/src/tracing/agent.cc
+++ b/src/tracing/agent.cc
@@ -2,6 +2,8 @@
 
 #include <sstream>
 #include <string>
+#include "tracing/node_trace_buffer.h"
+#include "tracing/node_trace_writer.h"
 
 #include "env-inl.h"
 
diff --git a/src/tracing/agent.h b/src/tracing/agent.h
index e781281712bbf6..59a7912631d75a 100644
--- a/src/tracing/agent.h
+++ b/src/tracing/agent.h
@@ -1,15 +1,16 @@
 #ifndef SRC_TRACING_AGENT_H_
 #define SRC_TRACING_AGENT_H_
 
+#include "libplatform/v8-tracing.h"
 #include "node_platform.h"
-#include "tracing/node_trace_buffer.h"
-#include "tracing/node_trace_writer.h"
 #include "uv.h"
 #include "v8.h"
 
 namespace node {
 namespace tracing {
 
+using v8::platform::tracing::TracingController;
+
 class Agent {
  public:
   Agent();