diff --git a/include/otbTensorflowSampler.hxx b/include/otbTensorflowSampler.hxx
index a552fea0b78e311ccd80c76ec6f9bbe63e7f06d0..5cf15beb11e238f70690053d85e4a8030cbb148d 100644
--- a/include/otbTensorflowSampler.hxx
+++ b/include/otbTensorflowSampler.hxx
@@ -199,8 +199,6 @@ TensorflowSampler<TInputImage, TVectorData>
             PixelType pix = it.Get();
             for (unsigned int band; band < pix.Size(); band++)
               {
-              std::cout << band << std::endl;
-
               if (pix[band] == m_NodataValue)
                 {
                 hasBeenSampled = false;
@@ -217,6 +215,20 @@ TensorflowSampler<TInputImage, TVectorData>
       } // Next input
       if (hasBeenSampled)
       {
+        // TODO: delete this
+        IndexType outIndex;
+        outIndex[0] = 0;
+        outIndex[1] = count * m_PatchSizes[0][1];
+        RegionType region(outIndex, m_PatchSizes[0]);
+        IteratorType it(m_OutputPatchImages[0], region);
+        for (it.GoToBegin(); !it.IsAtEnd(); ++it)
+          {
+          PixelType pix = it.Get();
+          for (unsigned int band; band < pix.Size(); band++)
+            if (pix[band] == m_NodataValue)
+              std::cout << band << std::endl;
+          }
+
         // Fill label
         labelIndex[1] = count;
         m_OutputLabelImage->SetPixel(labelIndex, labelPix);