diff --git a/tools/group_images.py b/tools/group_images.py
index 19435f6e5..e92aef54b 100644
--- a/tools/group_images.py
+++ b/tools/group_images.py
@@ -4,10 +4,12 @@
 import os
 import numpy as np
 
-from library.custom_logging import setup_logging
+from library.utils import setup_logging
+import logging
 
 # Set up logging
-log = setup_logging()
+setup_logging()
+log = logging.getLogger(__name__)
 
 class ImageProcessor:
 
diff --git a/tools/lcm_convert.py b/tools/lcm_convert.py
index 1fc39f69d..96d0d7788 100644
--- a/tools/lcm_convert.py
+++ b/tools/lcm_convert.py
@@ -1,12 +1,15 @@
 import argparse
 import torch
-from library.custom_logging import setup_logging
+import logging
+from library.utils import setup_logging
 from diffusers import StableDiffusionPipeline, StableDiffusionXLPipeline, LCMScheduler
 from library.sdxl_model_util import convert_diffusers_unet_state_dict_to_sdxl, sdxl_original_unet, save_stable_diffusion_checkpoint, _load_state_dict_on_device as load_state_dict_on_device
 from accelerate import init_empty_weights
 
 # Initialize logging
-logger = setup_logging()
+setup_logging()
+logger = logging.getLogger(__name__)
+
 
 def parse_command_line_arguments():
     argument_parser = argparse.ArgumentParser("lcm_convert")