diff --git a/stage_4_process.py b/stage_4_process.py
index f046ea9..b8e4382 100644
--- a/stage_4_process.py
+++ b/stage_4_process.py
@@ -498,7 +498,7 @@ def process_trial(
             volume_heartbeat_cpu, _, _ = torch.linalg.svd(
                 heartbeat_ts.cpu(), full_matrices=False
             )
-            volume_heartbeat = volume_heartbeat_cpu.to(heartbeat_ts.data, copy=True)
+            volume_heartbeat = volume_heartbeat_cpu.to(heartbeat_ts.device, copy=True)
             del volume_heartbeat_cpu
 
         volume_heartbeat = volume_heartbeat[:, 0]