diff --git a/train.py b/train.py
index ec9b148..8b6b2f9 100644
--- a/train.py
+++ b/train.py
@@ -537,7 +537,7 @@ def main(args):
         logging.info(f"{Fore.CYAN} * Training Text Encoder Only *{Style.RESET_ALL}")
         if text_encoder_lr_scale != 1:
             logging.warning(f"{Fore.YELLOW} * Ignoring text_encoder_lr_scale {text_encoder_lr_scale} and using the "
-                            f"Unet LR {curr_lr} for the text encoder instead.")
+                            f"Unet LR {curr_lr} for the text encoder instead *{Style.RESET_ALL}")
         params_to_train = itertools.chain(text_encoder.parameters())
     else:
         logging.info(f"{Fore.CYAN} * Training Text and Unet *{Style.RESET_ALL}")