diff --git a/packaging_lumbar_rootlets/run_inference_single_subject.py b/packaging_lumbar_rootlets/run_inference_single_subject.py
index ebb1bab..cdd0fa1 100644
--- a/packaging_lumbar_rootlets/run_inference_single_subject.py
+++ b/packaging_lumbar_rootlets/run_inference_single_subject.py
@@ -174,8 +174,9 @@ def main():
         tile_step_size=args.tile_step_size,     # changing it from 0.5 to 0.9 makes inference faster
         use_gaussian=True,      # applies gaussian noise and gaussian blur
         use_mirroring=False,    # test time augmentation by mirroring on all axes
-        perform_everything_on_device=True if args.use_gpu else False,
-        device=torch.device('cuda') if args.use_gpu else torch.device('cpu'),
+        perform_everything_on_device=True,
+        #device=torch.device('cuda') if args.use_gpu else torch.device('cpu'),
+        device=torch.device('mps'),
         verbose_preprocessing=False,
         allow_tqdm=True
     )