diff --git a/src/maxdiffusion/max_logging.py b/src/maxdiffusion/max_logging.py
index 2edb43f42..a3b3748d3 100644
--- a/src/maxdiffusion/max_logging.py
+++ b/src/maxdiffusion/max_logging.py
@@ -16,6 +16,11 @@
 
 """Stub for logging utilities. Right now just meant to avoid raw prints"""
 
+import logging
+
+logger = logging.getLogger(__name__)
+
 
 def log(user_str):
+  logger.info(user_str)
   print(user_str, flush=True)