diff --git a/Modules/Segmentation/Interactions/mitkMonaiLabel2DTool.cpp b/Modules/Segmentation/Interactions/mitkMonaiLabel2DTool.cpp
index d4c6213407..20427becea 100644
--- a/Modules/Segmentation/Interactions/mitkMonaiLabel2DTool.cpp
+++ b/Modules/Segmentation/Interactions/mitkMonaiLabel2DTool.cpp
@@ -1,76 +1,77 @@
 /*============================================================================
 
 The Medical Imaging Interaction Toolkit (MITK)
 
 Copyright (c) German Cancer Research Center (DKFZ)
 All rights reserved.
 
 Use of this source code is governed by a 3-clause BSD license that can be
 found in the LICENSE file.
 
 ============================================================================*/
 
 // MITK
 #include "mitkMonaiLabel2DTool.h"
 #include <mitkIOUtil.h>
 #include <mitkImageReadAccessor.h>
 #include <mitkSegTool2D.h>
 
 // us
 #include <usGetModuleContext.h>
 #include <usModule.h>
 #include <usModuleContext.h>
 #include <usModuleResource.h>
 #include <usServiceReference.h>
 
 namespace mitk
 {
   MITK_TOOL_MACRO(MITKSEGMENTATION_EXPORT, MonaiLabel2DTool, "MonaiLabel2D");
 }
 
 void mitk::MonaiLabel2DTool::Activated()
 {
   Superclass::Activated();
   this->SetLabelTransferScope(LabelTransferScope::AllLabels);
 }
 
 const char **mitk::MonaiLabel2DTool::GetXPM() const
 {
   return nullptr;
 }
 
 us::ModuleResource mitk::MonaiLabel2DTool::GetIconResource() const
 {
   us::Module *module = us::GetModuleContext()->GetModule();
   us::ModuleResource resource = module->GetResource("AI.svg");
   return resource;
 }
 
 const char *mitk::MonaiLabel2DTool::GetName() const
 {
   return "MONAI Label 2D";
 }
 
 void mitk::MonaiLabel2DTool::WriteImage(const Image *inputAtTimeStep, const std::string &inputImagePath) const
 {
   auto extendedImage = Image::New();
   std::array<unsigned int, 3> dim = {inputAtTimeStep->GetDimension(0), inputAtTimeStep->GetDimension(1), 1};
   mitk::PixelType pt = inputAtTimeStep->GetPixelType();
   extendedImage->Initialize(pt, 3, dim.data());
 
   ImageReadAccessor readAccessor(inputAtTimeStep);
   extendedImage->SetVolume(readAccessor.GetData());
 
   IOUtil::Save(extendedImage.GetPointer(), inputImagePath);
 }
 
 void mitk::MonaiLabel2DTool::WriteBackResults(LabelSetImage *previewImage,
                                               LabelSetImage *segResults,
                                               TimeStepType timeStep) const
  {
   if (segResults->GetTimeGeometry()->CountTimeSteps() > 1)
   {
-    mitkThrow() << "Invalid time geometry found while writing back segmentation";
+    mitkThrow() << "Invalid time geometry found while writing back segmentation. "
+                   " Expected static segmentation output from model.";
   }
   mitk::SegTool2D::WriteSliceToVolume(previewImage, this->GetWorkingPlaneGeometry(), segResults, timeStep, false);
  }
diff --git a/Modules/Segmentation/Interactions/mitkMonaiLabel3DTool.cpp b/Modules/Segmentation/Interactions/mitkMonaiLabel3DTool.cpp
index 3d155d18a4..b83a3363fa 100644
--- a/Modules/Segmentation/Interactions/mitkMonaiLabel3DTool.cpp
+++ b/Modules/Segmentation/Interactions/mitkMonaiLabel3DTool.cpp
@@ -1,66 +1,67 @@
 /*============================================================================
 
 The Medical Imaging Interaction Toolkit (MITK)
 
 Copyright (c) German Cancer Research Center (DKFZ)
 All rights reserved.
 
 Use of this source code is governed by a 3-clause BSD license that can be
 found in the LICENSE file.
 
 ============================================================================*/
 
 #include "mitkMonaiLabel3DTool.h"
 
 #include <mitkIOUtil.h>
 #include <usGetModuleContext.h>
 #include <usModule.h>
 #include <usModuleContext.h>
 #include <usModuleResource.h>
 #include <usServiceReference.h>
 #include <mitkImageReadAccessor.h>
 
 namespace mitk
 {
   MITK_TOOL_MACRO(MITKSEGMENTATION_EXPORT, MonaiLabel3DTool, "MonaiLabel3D");
 }
 
 void mitk::MonaiLabel3DTool::Activated()
 {
   Superclass::Activated();
   this->SetLabelTransferScope(LabelTransferScope::AllLabels);
 }
 
 const char **mitk::MonaiLabel3DTool::GetXPM() const
 {
   return nullptr;
 }
 
 us::ModuleResource mitk::MonaiLabel3DTool::GetIconResource() const
 {
   us::Module *module = us::GetModuleContext()->GetModule();
   us::ModuleResource resource = module->GetResource("AI.svg");
   return resource;
 }
 
 const char *mitk::MonaiLabel3DTool::GetName() const
 {
   return "MONAI Label 3D";
 }
 
 void mitk::MonaiLabel3DTool::WriteImage(const Image *inputAtTimeStep, const std::string &inputImagePath) const
 {
   IOUtil::Save(inputAtTimeStep, inputImagePath);
 }
 
 void mitk::MonaiLabel3DTool::WriteBackResults(LabelSetImage *previewImage,
                                               LabelSetImage *segResults,
                                               TimeStepType timeStep) const
 {
   if (segResults->GetTimeGeometry()->CountTimeSteps() > 1)
   {
-    mitkThrow() << "Invalid time geometry found while writing back segmentation";
+    mitkThrow() << "Invalid time geometry found while writing back segmentation. "
+                   " Expected static segmentation output from model.";
   }
   mitk::ImageReadAccessor newMitkImgAcc(segResults);
   previewImage->SetVolume(newMitkImgAcc.GetData(), timeStep);
 }