diff --git a/vall_e/inference.py b/vall_e/inference.py
index 332835e..7663d29 100755
--- a/vall_e/inference.py
+++ b/vall_e/inference.py
@@ -210,7 +210,7 @@ class TTS():
 				else:
 					raise Exception("!")
 				
-				text_list = [ cfg.tokenizer.decode( text ) for text in text_list ]
+				text_list = [ cfg.tokenizer.decode( text ).replace("   ", "_").replace(" ", "").replace("_", " ") for text in text_list ]
 				print( text_list )
 
 			return text_list[0]
diff --git a/vall_e/webui.py b/vall_e/webui.py
index 59b45fd..6032543 100644
--- a/vall_e/webui.py
+++ b/vall_e/webui.py
@@ -124,7 +124,7 @@ def do_inference_tts( progress=gr.Progress(track_tqdm=True), *args, **kwargs ):
 	parser = argparse.ArgumentParser(allow_abbrev=False)
 	# I'm very sure I can procedurally generate this list
 	parser.add_argument("--text", type=str, default=kwargs["text"])
-	parser.add_argument("--task", type=str, default=kwargs["task"])
+	parser.add_argument("--task", type=str, default="tts")
 	parser.add_argument("--references", type=str, default=kwargs["reference"])
 	parser.add_argument("--language", type=str, default="en")
 	parser.add_argument("--input-prompt-length", type=float, default=kwargs["input-prompt-length"])