diff --git a/src/serve.py b/src/serve.py
index 60b86ca..c666527 100644
--- a/src/serve.py
+++ b/src/serve.py
@@ -2,6 +2,7 @@ import gzip
 import json
 import logging
 
+from image_prediction import logger
 from image_prediction.config import Config
 from image_prediction.locations import CONFIG_FILE
 from image_prediction.pipeline import load_pipeline
@@ -15,7 +16,7 @@ PYINFRA_CONFIG = config.get_config()
 IMAGE_CONFIG = Config(CONFIG_FILE)
 
 # logging.getLogger().addHandler(logging.StreamHandler())
-logger = logging.getLogger(__name__)
+# logger = logging.getLogger(__name__)
 logger.setLevel(PYINFRA_CONFIG.logging_level_root)