diff --git a/tmva/tmva/src/MethodDNN.cxx b/tmva/tmva/src/MethodDNN.cxx
index 1c49a878eaa433da25e513334c345ba47ad60d23..901886df238db6af57bfe3d10ed2ab66dafdfec5 100644
--- a/tmva/tmva/src/MethodDNN.cxx
+++ b/tmva/tmva/src/MethodDNN.cxx
@@ -419,9 +419,9 @@ void TMVA::MethodDNN::ProcessOptions()
    fLayout = TMVA::MethodDNN::ParseLayoutString (fLayoutString);
    size_t inputSize = GetNVariables ();
    size_t outputSize = 1;
-   if (GetNTargets() != 0) {
+   if (fAnalysisType == Types::kRegression && GetNTargets() != 0) {
       outputSize = GetNTargets();
-   } else if (DataInfo().GetNClasses() > 2) {
+   } else if (fAnalysisType == Types::kMulticlass && DataInfo().GetNClasses() >= 2) {
       outputSize = DataInfo().GetNClasses();
    }