diff --git a/Modules/Applications/AppClassification/app/otbSampleAugmentation.cxx b/Modules/Applications/AppClassification/app/otbSampleAugmentation.cxx
index eed67c8fb8d84102d4a3459f1571d4fcae23aa31..b2fcb1deb64251194777565cf70bf668dea3efd4 100644
--- a/Modules/Applications/AppClassification/app/otbSampleAugmentation.cxx
+++ b/Modules/Applications/AppClassification/app/otbSampleAugmentation.cxx
@@ -248,14 +248,14 @@ private:
     {
     otbAppLogINFO("Augmentation strategy : jitter");
     filter->SetStrategy(FilterType::Strategy::Jitter);
-    filter->SetStdFactor(this->GetParameterFloat("stdfactor"));
+    filter->SetStdFactor(this->GetParameterFloat("strategy.jitter.stdfactor"));
     }
     break;
     case 2:
     {
     otbAppLogINFO("Augmentation strategy : smote");
     filter->SetStrategy(FilterType::Strategy::Smote);
-    filter->SetSmoteNeighbors(this->GetParameterInt("neighbors"));
+    filter->SetSmoteNeighbors(this->GetParameterInt("strategy.smote.neighbors"));
     }
     break;
     }
diff --git a/Modules/Applications/AppClassification/include/otbSampleAugmentationFilter.txx b/Modules/Applications/AppClassification/include/otbSampleAugmentationFilter.txx
index 41590a0109cf0242ca70bf6e3404c9242acfdd85..612e0e22f559a7e38be76bde767da59a34b83bd1 100644
--- a/Modules/Applications/AppClassification/include/otbSampleAugmentationFilter.txx
+++ b/Modules/Applications/AppClassification/include/otbSampleAugmentationFilter.txx
@@ -80,7 +80,7 @@ SampleAugmentationFilter
 {
 
   OGRDataSourcePointerType inputDS = dynamic_cast<OGRDataSourceType*>(this->itk::ProcessObject::GetInput(0));
-  auto outputDS = static_cast<ogr::DataSource *>(this->itk::ProcessObject::GetOutput(1));
+  auto outputDS = static_cast<ogr::DataSource *>(this->itk::ProcessObject::GetOutput(0));
   auto inSamples = this->extractSamples(inputDS, m_Layer,
                                         m_ClassFieldName,
                                         m_Label,