1 files changed, 3 insertions, 0 deletions

diff --git a/collector/perf_linux.go b/collector/perf_linux.go
index bb8716a..6d19683 100644
--- a/collector/perf_linux.go
+++ b/collector/perf_linux.go
@@ -135,18 +135,21 @@ func NewPerfCollector(logger log.Logger) (Collector, error) {
                         return nil, err
                 }
                 collector.perfHwProfilers[cpu] = &hwProf
+                collector.hwProfilerCPUMap[&hwProf] = cpu
 
                 swProf := perf.NewSoftwareProfiler(-1, cpu)
                 if err := swProf.Start(); err != nil {
                         return nil, err
                 }
                 collector.perfSwProfilers[cpu] = &swProf
+                collector.swProfilerCPUMap[&swProf] = cpu
 
                 cacheProf := perf.NewCacheProfiler(-1, cpu)
                 if err := cacheProf.Start(); err != nil {
                         return nil, err
                 }
                 collector.perfCacheProfilers[cpu] = &cacheProf
+                collector.cacheProfilerCPUMap[&cacheProf] = cpu
         }
 
         collector.desc = map[string]*prometheus.Desc{