1 files changed, 3 insertions, 3 deletions
diff --git a/Omni/Log.py b/Omni/Log.py
index ee4a050..5b3a618 100644
--- a/Omni/Log.py
+++ b/Omni/Log.py
@@ -13,7 +13,7 @@ class LowerFormatter(logging.Formatter):
         return super().format(record)
 
 
-def setup(level: int = logging.INFO) -> logging.Logger:
+def setup(logger: logging.Logger, level: int = logging.INFO) -> logging.Logger:
     """Run this in your `main()` function."""
     logging.basicConfig(
         level=level,
@@ -22,7 +22,6 @@ def setup(level: int = logging.INFO) -> logging.Logger:
     logging.addLevelName(logging.DEBUG, "dbug")
     logging.addLevelName(logging.ERROR, "fail")
     logging.addLevelName(logging.INFO, "info")
-    logger = logging.getLogger(__name__)
     formatter = LowerFormatter()
     handler = logging.StreamHandler()
     handler.setFormatter(formatter)
@@ -32,5 +31,6 @@ def setup(level: int = logging.INFO) -> logging.Logger:
 
 def main() -> None:
     """Entrypoint to test that this kinda works."""
-    logger = setup()
+    logger = logging.getLogger(__name__)
+    setup(logger)
     logger.debug("i am doing testing")