diff --git a/DataFormats/Detectors/TPC/CMakeLists.txt b/DataFormats/Detectors/TPC/CMakeLists.txt
index b8b93c308e85d..81b1d5efad59a 100644
--- a/DataFormats/Detectors/TPC/CMakeLists.txt
+++ b/DataFormats/Detectors/TPC/CMakeLists.txt
@@ -34,6 +34,7 @@ o2_add_library(
                         O2::ReconstructionDataFormats
                         O2::CommonDataFormat
                         O2::Headers
+                        O2::DataSampling
                         O2::Algorithm)
 
 o2_target_root_dictionary(
diff --git a/DataFormats/Detectors/TPC/include/DataFormatsTPC/WorkflowHelper.h b/DataFormats/Detectors/TPC/include/DataFormatsTPC/WorkflowHelper.h
index 30b40ed70b9c7..f4a318bc30101 100644
--- a/DataFormats/Detectors/TPC/include/DataFormatsTPC/WorkflowHelper.h
+++ b/DataFormats/Detectors/TPC/include/DataFormatsTPC/WorkflowHelper.h
@@ -21,6 +21,7 @@
 #include "Framework/DataRefUtils.h"
 #include <Framework/InputRecord.h>
 #include "Framework/InputRecordWalker.h"
+#include "DataSampling/DataSamplingHeader.h"
 #include "DataFormatsTPC/TrackTPC.h"
 #include "SimulationDataFormat/MCCompLabel.h"
 #include "SimulationDataFormat/MCTruthContainer.h"
@@ -80,7 +81,7 @@ static auto getWorkflowTPCInput(o2::framework::ProcessingContext& pc, int verbos
     if (do_digits) {
       std::fill(inputDigitsMCIndex.begin(), inputDigitsMCIndex.end(), -1);
     }
-    for (auto const& ref : o2::framework::InputRecordWalker(pc.inputs(), filter)) {
+    for (auto const& ref : o2::framework::InputRecordWalker<o2::utilities::DataSamplingHeader>(pc.inputs(), filter)) {
       auto const* sectorHeader = o2::framework::DataRefUtils::getHeader<TPCSectorHeader*>(ref);
       if (sectorHeader == nullptr) {
         // FIXME: think about error policy
@@ -127,7 +128,7 @@ static auto getWorkflowTPCInput(o2::framework::ProcessingContext& pc, int verbos
       {"check", o2::framework::ConcreteDataTypeMatcher{o2::header::gDataOriginTPC, "CLUSTERNATIVE"}, o2::framework::Lifetime::Timeframe},
     };
     unsigned long recvMask = 0;
-    for (auto const& ref : o2::framework::InputRecordWalker(pc.inputs(), filter)) {
+    for (auto const& ref : o2::framework::InputRecordWalker<o2::utilities::DataSamplingHeader>(pc.inputs(), filter)) {
       auto const* sectorHeader = o2::framework::DataRefUtils::getHeader<TPCSectorHeader*>(ref);
       if (sectorHeader == nullptr) {
         throw std::runtime_error("sector header missing on header stack");