diff --git a/lib/core/common.py b/lib/core/common.py
index 0852014fd..0e5786228 100644
--- a/lib/core/common.py
+++ b/lib/core/common.py
@@ -700,7 +700,7 @@ def setColor(message, bold=False):
     level = extractRegexResult(r"\A\s*\[(?P<result>[A-Z ]+)\]", message)
 
     if hasattr(LOGGER_HANDLER, "level_map"):
-        _ = LOGGER_HANDLER.level_map.get(logging._levelNames.get(level))
+        _ = LOGGER_HANDLER.level_map.get(logging.getLevelName(level))
         if _:
             background, foreground, bold = _
             retVal = colored(message, color=foreground, on_color="on_%s" % background if background else None, attrs=("bold",) if bold else None)
diff --git a/lib/core/log.py b/lib/core/log.py
index 1e3a883a7..13d0bc035 100644
--- a/lib/core/log.py
+++ b/lib/core/log.py
@@ -21,6 +21,8 @@ LOGGER = logging.getLogger("sqlmapLog")
 try:
     import ctypes
     LOGGER_HANDLER = ColorizingStreamHandler(sys.stdout)
+    LOGGER_HANDLER.level_map[logging.getLevelName("TRAFFIC OUT")] = (None, "cyan", True)
+    LOGGER_HANDLER.level_map[logging.getLevelName("TRAFFIC IN")] = (None, "grey", True)
 except ImportError:
     LOGGER_HANDLER = logging.StreamHandler(sys.stdout)