diff --git a/src/llmtuner/extras/misc.py b/src/llmtuner/extras/misc.py
index 4f123e14..9a50e369 100644
--- a/src/llmtuner/extras/misc.py
+++ b/src/llmtuner/extras/misc.py
@@ -5,8 +5,8 @@ import torch
 from typing import TYPE_CHECKING, Any, Dict, Optional, Tuple
 from transformers import InfNanRemoveLogitsProcessor, LogitsProcessorList
 
-import logging
-logger = logging.getLogger(__name__)
+from llmtuner.extras.logging import get_logger
+logger = get_logger(__name__)
 
 try:
     from transformers.utils import (