diff --git a/include/otbTensorflowMultisourceModelBase.h b/include/otbTensorflowMultisourceModelBase.h
index dc025bcb66c48ddafa9fed8ed486c53762ddebe7..0929f9494fed952c47c36f9703df8a28aca94916 100644
--- a/include/otbTensorflowMultisourceModelBase.h
+++ b/include/otbTensorflowMultisourceModelBase.h
@@ -72,7 +72,7 @@ public:
 
   /** Standard class typedefs. */
   typedef TensorflowMultisourceModelBase             Self;
-  typedef itk::ProcessObject                         Superclass;
+  typedef itk::ImageToImageFilter<TInputImage, TOutputImage> Superclass;
   typedef itk::SmartPointer<Self>                    Pointer;
   typedef itk::SmartPointer<const Self>              ConstPointer;
 
diff --git a/include/otbTensorflowMultisourceModelBase.hxx b/include/otbTensorflowMultisourceModelBase.hxx
index baaa3cba305014f69b81f1cafd540c8dc98015cf..573aa9ca714f7a163076cc3c9ec4466998e60db5 100644
--- a/include/otbTensorflowMultisourceModelBase.hxx
+++ b/include/otbTensorflowMultisourceModelBase.hxx
@@ -21,6 +21,8 @@ TensorflowMultisourceModelBase<TInputImage, TOutputImage>
 ::TensorflowMultisourceModelBase()
  {
   m_Session = nullptr;
+  Superclass::SetCoordinateTolerance(itk::NumericTraits<double>::max() );
+  Superclass::SetDirectionTolerance(itk::NumericTraits<double>::max() );
  }
 
 template <class TInputImage, class TOutputImage>