diff --git a/nodes.py b/nodes.py
index 39d8d67..2076c9f 100644
--- a/nodes.py
+++ b/nodes.py
@@ -909,6 +909,7 @@ class FluxTrainEnd:
     RETURN_NAMES = ("lora_name", "metadata", "lora_path",)
     FUNCTION = "endtrain"
     CATEGORY = "FluxTrainer"
+    OUTPUT_NODE = True
 
     def endtrain(self, network_trainer, save_state):
         with torch.inference_mode(False):