#include "util/session.h"
 #include "util/tool.h"
 #include "util/debug.h"
+#include "util/build-id.h"
 
 #include "util/parse-options.h"
 
                return f(tool, event, sample, evsel, machine);
        }
 
+       build_id__mark_dso_hit(tool, event, sample, evsel, machine);
+
        return perf_event__repipe_synth(tool, event, machine);
 }
 
        sample_sw.time   = sample->time;
        perf_event__synthesize_sample(event_sw, evsel->attr.sample_type,
                                      &sample_sw, false);
+       build_id__mark_dso_hit(tool, event_sw, &sample_sw, evsel, machine);
        return perf_event__repipe(tool, event_sw, &sample_sw, machine);
 }
 
 
        signal(SIGINT, sig_handler);
 
-       if (inject->build_ids) {
-               inject->tool.sample       = perf_event__inject_buildid;
+       if (inject->build_ids || inject->sched_stat) {
                inject->tool.mmap         = perf_event__repipe_mmap;
                inject->tool.fork         = perf_event__repipe_fork;
                inject->tool.tracing_data = perf_event__repipe_tracing_data;
        if (session == NULL)
                return -ENOMEM;
 
-       if (inject->sched_stat) {
+       if (inject->build_ids) {
+               inject->tool.sample = perf_event__inject_buildid;
+       } else if (inject->sched_stat) {
                struct perf_evsel *evsel;
 
                inject->tool.ordered_samples = true;
 
 #include "session.h"
 #include "tool.h"
 
-static int build_id__mark_dso_hit(struct perf_tool *tool __maybe_unused,
-                                 union perf_event *event,
-                                 struct perf_sample *sample __maybe_unused,
-                                 struct perf_evsel *evsel __maybe_unused,
-                                 struct machine *machine)
+int build_id__mark_dso_hit(struct perf_tool *tool __maybe_unused,
+                          union perf_event *event,
+                          struct perf_sample *sample __maybe_unused,
+                          struct perf_evsel *evsel __maybe_unused,
+                          struct machine *machine)
 {
        struct addr_location al;
        u8 cpumode = event->header.misc & PERF_RECORD_MISC_CPUMODE_MASK;
 
 
 char *dso__build_id_filename(struct dso *self, char *bf, size_t size);
 
+int build_id__mark_dso_hit(struct perf_tool *tool, union perf_event *event,
+                          struct perf_sample *sample, struct perf_evsel *evsel,
+                          struct machine *machine);
+
 #endif