diff --git a/Modules/Applications/AppClassification/app/otbTrainImagesRegression.cxx b/Modules/Applications/AppClassification/app/otbTrainImagesRegression.cxx
index 19177493327f198645a54d0d582843e3cad4e534..fbfab6df873d7339727250e171eba74ddd46d4db 100644
--- a/Modules/Applications/AppClassification/app/otbTrainImagesRegression.cxx
+++ b/Modules/Applications/AppClassification/app/otbTrainImagesRegression.cxx
@@ -354,6 +354,11 @@ protected:
     ShareParameter( "ram", "polystat.ram" );
     Connect( "select.ram", "polystat.ram" );
     Connect( "extraction.ram", "polystat.ram" );
+    
+    ShareParameter( "elev", "polystat.elev");
+    Connect( "select.elev", "polystat.elev");
+    if (!IsParameterEnabled("io.vd") || !HasValue("io.vd"))
+      Connect( "imageEnvelope.elev", "polystat.elev");
   }
 
   /** Init Learning applications and share learning parameters */