diff --git a/scraibe/cli.py b/scraibe/cli.py
index e4eeaad..323a1b1 100644
--- a/scraibe/cli.py
+++ b/scraibe/cli.py
@@ -100,6 +100,7 @@ def str2bool(string):
                     'whisper_type':arg_dict.pop("whisper_type"),
                     'dia_model': arg_dict.pop("diarization_directory"),
                     'use_auth_token': arg_dict.pop("hf_token"),
+                    'device': arg_dict.pop('inference_device')
                     }
 
     if arg_dict["whisper_model_directory"]: