diff --git a/whisperlivekit/audio_processor.py b/whisperlivekit/audio_processor.py
index f2cd342..7b7b3fd 100644
--- a/whisperlivekit/audio_processor.py
+++ b/whisperlivekit/audio_processor.py
@@ -483,7 +483,7 @@ class AudioProcessor:
                     status=response_status,
                     lines=lines,
                     buffer_transcription=buffer_transcription.text.strip(),
-                    buffer_diarization=buffer_diarization.strip(),
+                    buffer_diarization=buffer_diarization,
                     remaining_time_transcription=state.remaining_time_transcription,
                     remaining_time_diarization=state.remaining_time_diarization if self.args.diarization else 0
                 )
diff --git a/whisperlivekit/simul_whisper/backend.py b/whisperlivekit/simul_whisper/backend.py
index c1ed4e8..cdb4303 100644
--- a/whisperlivekit/simul_whisper/backend.py
+++ b/whisperlivekit/simul_whisper/backend.py
@@ -22,11 +22,9 @@ try:
     HAS_MLX_WHISPER = True
 except ImportError:
     if platform.system() == "Darwin" and platform.machine() == "arm64":
-        print(f"""
-            {"="*50}
-            MLX Whisper not found but you are on Apple Silicon. Consider installing mlx-whisper for better performance: pip install mlx-whisper
-            {"="*50}
-            """)
+        print(f"""{"="*50}
+MLX Whisper not found but you are on Apple Silicon. Consider installing mlx-whisper for better performance: pip install mlx-whisper
+{"="*50}""")
     HAS_MLX_WHISPER = False
 if HAS_MLX_WHISPER:
     HAS_FASTER_WHISPER = False
diff --git a/whisperlivekit/web/live_transcription.css b/whisperlivekit/web/live_transcription.css
index 9b2b520..363c01c 100644
--- a/whisperlivekit/web/live_transcription.css
+++ b/whisperlivekit/web/live_transcription.css
@@ -483,7 +483,6 @@ label {
 
 .buffer_diarization {
   color: var(--label-dia-text);
-  margin-left: 4px;
 }
 
 .buffer_transcription {
@@ -600,8 +599,6 @@ label {
 .label_language {
   background-color: var(--chip-bg);
   margin-bottom: 0px;
-  margin-top: 5px;
-  height: 18.5px;
   border-radius: 100px;
   padding: 2px 8px;
   margin-left: 10px;