diff --git a/src/main.rs b/src/main.rs
index 43b6cb95..f0ea3840 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -222,7 +222,7 @@ fn model_task(
             token_counter.prompt_tokens = tokens.len();
             token_counter.total_tokens = tokens.len();
 
-            log::trace!("{prompt}");
+            log::trace!("{}", String::from_utf8(remain).unwrap_or_default());
 
             for _ in 0..max_tokens {
                 if token_sender.is_disconnected() {
@@ -267,7 +267,7 @@ fn model_task(
             let _ = token_sender.send(Token::Stop(FinishReason::Length, token_counter));
         }
 
-        print!("\n\n");
+        println!("[DONE]");
 
         if let Ok(back) = state.back() {
             if embedding {