1 files changed, 3 insertions, 2 deletions
diff --git a/Omni/Llamacpp.py b/Omni/Llamacpp.py
index bf44d5e..c7e9078 100755
--- a/Omni/Llamacpp.py
+++ b/Omni/Llamacpp.py
@@ -9,7 +9,7 @@ sure llama-cpp still works in case I need/want to switch at some point.
 
 # : out llamacpp-test
 # : run llama-cpp
-
+import logging
 import Omni.App as App
 import Omni.Log as Log
 import Omni.Test as Test
@@ -29,7 +29,8 @@ class TestLlamaCpp(unittest.TestCase):
 def main() -> None:
     """Entrypoint."""
     if sys.argv[1] == "test":
-        Log.setup()
+        logger = logging.getLogger(__name__)
+        Log.setup(logger)
         Test.run(App.Area.Test, [TestLlamaCpp])
     else:
         sys.exit(0)