diff --git a/terratorch/cli_tools.py b/terratorch/cli_tools.py
index a00f2ad3..e8dceae1 100644
--- a/terratorch/cli_tools.py
+++ b/terratorch/cli_tools.py
@@ -370,7 +370,6 @@ def add_arguments_to_parser(self, parser: LightningArgumentParser) -> None:
 
     def instantiate_classes(self) -> None:
 
-        super().instantiate_classes()
         # get the predict_output_dir. Depending on the value of run, it may be in the subcommand
         try:
             config = self.config.predict
@@ -398,6 +397,7 @@ def instantiate_classes(self) -> None:
             custom_modules_path = os.getenv("TERRATORCH_CUSTOM_MODULE_PATH", None)
 
         import_custom_modules(custom_modules_path)
+        super().instantiate_classes()
 
 def build_lightning_cli(
     args: ArgsType = None,