2 files changed, 5 insertions, 5 deletions

diff --git a/litmus/litmus.c b/litmus/litmus.c
index 703360c68609..db5ce0e9c76e 100644
--- a/litmus/litmus.c
+++ b/litmus/litmus.c
@@ -26,6 +26,11 @@
 #include <litmus/affinity.h>
 #endif
 
+#ifdef CONFIG_SCHED_LITMUS_TRACEPOINT
+#define CREATE_TRACE_POINTS
+#include <trace/events/litmus.h>
+#endif
+
 /* Number of RT tasks that exist in the system */
 atomic_t rt_task_count          = ATOMIC_INIT(0);
 
diff --git a/litmus/sched_task_trace.c b/litmus/sched_task_trace.c
index fb4eddf1d7bc..c154ec48be41 100644
--- a/litmus/sched_task_trace.c
+++ b/litmus/sched_task_trace.c
@@ -15,11 +15,6 @@
 #include <litmus/feather_trace.h>
 #include <litmus/ftdev.h>
 
-#ifdef CONFIG_SCHED_LITMUS_TRACEPOINT
-#define CREATE_TRACE_POINTS
-#include <trace/events/litmus.h>
-#endif
-
 #define NO_EVENTS               (1 << CONFIG_SCHED_TASK_TRACE_SHIFT)
 
 #define now() litmus_clock()