diff --git a/utils/logger.py b/utils/logger.py
index 4d7e5ce..3461c58 100644
--- a/utils/logger.py
+++ b/utils/logger.py
@@ -38,5 +38,5 @@ class CustomFormatter(logging.Formatter):
 
         return super().format(record)
 
-    def highlight_args(self, message, args, color):
+    def highlight_message(self, message, args, color):
         return f"{color}{message}{RESET}"