diff --git a/nanocode.py b/nanocode.py
index d1f7ab9..090d673 100644
--- a/nanocode.py
+++ b/nanocode.py
@@ -222,8 +222,8 @@ class RLMCodingProgram(PrecompiledProgram):
         )
         self.agent.set_lm(self.lm)
         print(f"{BLUE}CONFIG WAS SET: {self.config}{RESET}")
-        print(f"{BLUE}LM WAS SET: {self.agent.get_lm()}{RESET}")
-        print(f"{BLUE}SUB LM WAS SET: {self.agent.sub_lm}{RESET}")
+        print(f"{BLUE}LM WAS SET: {self.agent.get_lm().model}{RESET}")
+        print(f"{BLUE}SUB LM WAS SET: {self.agent.sub_lm.model}{RESET}")
 
     def forward(self, task: str) -> str:
         if not task: