diff --git a/lollms/personality.py b/lollms/personality.py
index 94677f9..46cee97 100644
--- a/lollms/personality.py
+++ b/lollms/personality.py
@@ -729,7 +729,7 @@ def process(self, text:str, message_type:MSG_OPERATION_TYPE, callback=None, show
             self.bot_says = bot_says
             return True
 
-    def generate_with_images(self, prompt, images, max_size, temperature = None, top_k = None, top_p=None, repeat_penalty=None, repeat_last_n=None, callback=None, debug=False, show_progress=False ):
+    def generate_with_images(self, prompt, images, max_size=None, temperature = None, top_k = None, top_p=None, repeat_penalty=None, repeat_last_n=None, callback=None, debug=False, show_progress=False ):
         ASCIIColors.info("Text generation started: Warming up")
         self.nb_received_tokens = 0
         self.bot_says = ""
@@ -2427,7 +2427,7 @@ def generate_code(self, prompt, images=[], max_size = None,  temperature = None,
                         else:
                             code +="\n"+ "\n".join(codes[-1]["content"].split("\n"))
             else:
-                code = "\n".join(codes[-1]["content"].split("\n"))
+                code = codes[-1]["content"]
 
             return code
         else: