diff --git a/src/mednet/libs/common/scripts/train_analysis.py b/src/mednet/libs/common/scripts/train_analysis.py
index bf3efe885b7b9f374dc96bd6035c46881b2f279b..65350228a77528575802ed1191adab96c287688a 100644
--- a/src/mednet/libs/common/scripts/train_analysis.py
+++ b/src/mednet/libs/common/scripts/train_analysis.py
@@ -6,7 +6,7 @@ import pathlib
 import typing
 
 import click
-from clapper.click import ResourceOption, verbosity_option
+from clapper.click import verbosity_option
 from clapper.logging import setup
 
 from .click import ConfigCommand
@@ -220,6 +220,7 @@ def _create_figures(
         writable=True,
         path_type=pathlib.Path,
     ),
+    default="results",
 )
 @verbosity_option(logger=logger, expose_value=False)
 def train_analysis(
diff --git a/src/mednet/libs/segmentation/scripts/cli.py b/src/mednet/libs/segmentation/scripts/cli.py
index 82aaed41dff680f167236d07e9f67495447c88e3..52ce017f997cc9bc0678378a671aea0da97cde9e 100644
--- a/src/mednet/libs/segmentation/scripts/cli.py
+++ b/src/mednet/libs/segmentation/scripts/cli.py
@@ -2,6 +2,8 @@
 #
 # SPDX-License-Identifier: GPL-3.0-or-later
 
+import importlib
+
 import click
 from clapper.click import AliasedGroup
 
@@ -38,4 +40,9 @@ segmentation.add_command(database.database)
 # segmentation.add_command(predict.predict)
 # segmentation.add_command(significance.significance)
 segmentation.add_command(train.train)
-# segmentation.add_command(train_analysis.train_analysis)
+segmentation.add_command(
+    importlib.import_module(
+        "mednet.libs.common.scripts.train_analysis",
+        package=__name__,
+    ).train_analysis,
+)