diff --git a/ml/actions/labeler.py b/ml/actions/labeler.py
index 10278ae..d3b7afd 100644
--- a/ml/actions/labeler.py
+++ b/ml/actions/labeler.py
@@ -240,12 +240,12 @@ def print_label_distribution(labels: np.ndarray) -> Dict[str, int]:
         dist[action.value] = count
 
     total = len(labels)
-    log.info(f"\n{'Action':<25} {'Count':>8} {'Pct':>8}")
-    log.info("─" * 43)
+    logger.info(f"\n{'Action':<25} {'Count':>8} {'Pct':>8}")
+    logger.info("─" * 43)
     for action_name, count in sorted(dist.items(), key=lambda x: -x[1]):
         pct = (count / total * 100) if total > 0 else 0
-        log.info(f"  {action_name:<23} {count:>8} {pct:>7.1f}%")
-    log.info(f"{'─' * 43}")
-    log.info(f"  {'TOTAL':<23} {total:>8}")
+        logger.info(f"  {action_name:<23} {count:>8} {pct:>7.1f}%")
+    logger.info(f"{'─' * 43}")
+    logger.info(f"  {'TOTAL':<23} {total:>8}")
 
     return dist