diff --git a/audio.py b/audio.py
index f289850..537de40 100644
--- a/audio.py
+++ b/audio.py
@@ -438,6 +438,7 @@ class XttsNode:
             "--folder", kwargs.get('output_path', "."),
             "--lang", kwargs.get('lang', 'en'),
             "--speaker", kwargs.get('speaker_wav', ''),
+            "--device", kwargs.get('device', 'cuda:0'),
         ]
         # write the command to "temp_command.txt" in the same folder as this file is in
         with open(os.path.join(this_file_path, "temp_command.txt"), "w") as f:
@@ -446,7 +447,8 @@ class XttsNode:
         # execute the command in its own full porcess and recieve the output
         try:
             # Capture both stdout and stderr
-            output = subprocess.check_output(command, shell=True, stderr=subprocess.STDOUT)
+            #output = subprocess.check_output(command, shell=True, stderr=subprocess.STDOUT)
+            output = subprocess.check_output(command, stderr=subprocess.STDOUT)
             generated = output.decode("utf-8")
 
             if "Generated files: (" in generated:
diff --git a/modules/xttscli.py b/modules/xttscli.py
index 9a9feb8..45715da 100644
--- a/modules/xttscli.py
+++ b/modules/xttscli.py
@@ -128,7 +128,7 @@ def main():
     parser.add_argument('--text', help="Text to synthesize.")
     parser.add_argument('--file', help="File containing text to synthesize (one text per line).")
     parser.add_argument('--lang', default='en', help="Language for the TTS.")
-    parser.add_argument('--output', help="Path to save the output TTS file.")
+    parser.add_argument('--folder', help="Path to save the output TTS file.")
     parser.add_argument('--device', default='cuda:0', help="Device to run the model on. Default: cuda:0")
     parser.add_argument('--test', action='store_true', help="Run a test TTS using the test.speech file.")
 
@@ -145,9 +145,9 @@ def main():
                 raise FileNotFoundError(f"The test file '{default_test_file_path}' does not exist. Please create it.")
 
             args.file = default_test_file_path
-            args.output = os.path.join(script_dir, 'test_output.wav')
+            args.folder = os.path.join(script_dir, 'test_output.wav')
             print(f"Test file: {args.file}")
-            print(f"Output will be saved to: {args.output}")
+            print(f"Output will be saved to: {args.folder}")
 
         # Handle the --file argument
         if args.file:
@@ -172,7 +172,7 @@ def main():
                     device=args.device
                 )
 
-                output_file = f"{args.output}_{i}.wav" if args.output else f'./output_{i}.wav'
+                output_file = f"{args.folder}_{i}.wav" if args.folder else f'./output_{i}.wav'
                 try:
                     os.rename(output_path, output_file)
                 except Exception as e:
@@ -192,7 +192,7 @@ def main():
                 device=args.device
             )
 
-            output_file = args.output if args.output else './output.wav'
+            output_file = args.folder if args.folder else './output.wav'
             try:
                 os.rename(output_path, output_file)
             except Exception as e: