diff --git a/RTCP/Cobalt/GPUProc/src/Flagger.cc b/RTCP/Cobalt/GPUProc/src/Flagger.cc
index 31d3e92c27a4a00fd89d677d73898997ca3831e2..57272e2e3800f9d55dd055e8b1083a34ed4fc012 100644
--- a/RTCP/Cobalt/GPUProc/src/Flagger.cc
+++ b/RTCP/Cobalt/GPUProc/src/Flagger.cc
@@ -30,7 +30,7 @@ namespace LOFAR
     void Flagger::convertFlagsToChannelFlags(
       MultiDimArray<LOFAR::SparseSet<unsigned>, 1>const &inputFlags,
       MultiDimArray<SparseSet<unsigned>, 1>& flagsPerChannel,
-      const unsigned nrSamplesPerChannel,
+      const unsigned nrSamples,
       const unsigned nrChannels,
       const ssize_t nrPrefixedSamples)
     {
@@ -39,6 +39,7 @@ namespace LOFAR
       // If nrChannels == 1, we do not expect nrPrefixedSamples
       ASSERT(nrChannels > 1 || nrPrefixedSamples == 0);
 
+      unsigned nrSamplesPerChannel = nrSamples / nrChannels;
       unsigned log2NrChannels = log2(nrChannels);
 
       // Convert the flags per sample to flags per channel
diff --git a/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/BeamFormerPreprocessingStep.cc b/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/BeamFormerPreprocessingStep.cc
index 0021057f0ea845255a3a3cbc1da1660f2f524a43..e85db40772492a8a63dfd4eabc20818ef7389670 100644
--- a/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/BeamFormerPreprocessingStep.cc
+++ b/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/BeamFormerPreprocessingStep.cc
@@ -124,7 +124,7 @@ namespace LOFAR
       Flagger::convertFlagsToChannelFlags(
         input.inputFlags,
         flagsPerChannel,
-        ps.settings.blockSize / ps.settings.beamFormer.nrDelayCompensationChannels,
+        ps.settings.blockSize,
         ps.settings.beamFormer.nrDelayCompensationChannels,
         0);
 
diff --git a/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/CorrelatorStep.cc b/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/CorrelatorStep.cc
index b1ed83fe4b42817c68ce961e57bf593191e67d51..b58796880974d8c4665ccf843e7c61bb5ba37269 100644
--- a/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/CorrelatorStep.cc
+++ b/RTCP/Cobalt/GPUProc/src/cuda/SubbandProcs/CorrelatorStep.cc
@@ -89,7 +89,7 @@ namespace LOFAR
       Cobalt::Flagger::convertFlagsToChannelFlags(
         inputFlags,
         flagsPerChannel,
-        parset.settings.correlator.nrSamplesPerBlock,
+        parset.settings.blockSize,
         parset.settings.correlator.nrChannels,
         parset.settings.correlator.nrChannels == 1 ? 0 : NR_TAPS - 1);
 
diff --git a/RTCP/Cobalt/GPUProc/test/SubbandProcs/tCorrelatorStep.cc b/RTCP/Cobalt/GPUProc/test/SubbandProcs/tCorrelatorStep.cc
index c23a63ec6812c688643cb6c76e184e2728677b3f..15a6d205df0284e6afd7b2fe4da4b81e14310086 100644
--- a/RTCP/Cobalt/GPUProc/test/SubbandProcs/tCorrelatorStep.cc
+++ b/RTCP/Cobalt/GPUProc/test/SubbandProcs/tCorrelatorStep.cc
@@ -79,7 +79,7 @@ TEST(convertFlagsToChannelFlags)
   CorrelatorStep::Flagger::convertFlagsToChannelFlags(
     inputFlags,
     flagsPerChannel,
-    parset.settings.correlator.nrSamplesPerBlock,
+    parset.settings.blockSize,
     parset.settings.correlator.nrChannels,
     NR_TAPS - 1
     );