diff --git a/Code/Markov/otbMRFEnergyGaussianClassification.h b/Code/Markov/otbMRFEnergyGaussianClassification.h
index d374b0b08a4746c0c3c0ff312edebc9f52cbaeb0..800b04cd35245eeb4054ddb3ceef775660de0a33 100644
--- a/Code/Markov/otbMRFEnergyGaussianClassification.h
+++ b/Code/Markov/otbMRFEnergyGaussianClassification.h
@@ -19,6 +19,7 @@
 #define _otbMRFEnergyGaussianClassification_h
 
 #include "otbMRFEnergy.h"
+#include "otbMath.h"
 
 #define M_SQUARE(x) ((x)*(x))
 
diff --git a/Code/Markov/otbMarkovClassificationFilter.txx b/Code/Markov/otbMarkovClassificationFilter.txx
index 3c0771ea5741ae98ce34119a8d302a7da10017a2..3f57c09b63e9ab6473d2f38b876a5c643c8b722e 100644
--- a/Code/Markov/otbMarkovClassificationFilter.txx
+++ b/Code/Markov/otbMarkovClassificationFilter.txx
@@ -91,7 +91,7 @@ MarkovClassificationFilter<TInputImage, TClassifiedImage>
     {
       return 0;
     }
-  return static_cast<const TrainingImageType * >
+  return static_cast<TrainingImageType * >
     (this->itk::ProcessObject::GetInput(1) );
 }