diff --git a/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d.py b/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d.py
index c98b36c..6eaec76 100644
--- a/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d.py
+++ b/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d.py
@@ -379,7 +379,7 @@ def predict(self, inputs, is_right_disparity=False):
         else:
             disp1_raw, disp2_raw = torch.split(inputs["disparities"], 1, dim=1)
 
-        K = inputs["intrinsics"][:, 0]
+        K = inputs["intrinsics"][:, 0, 0]
         intrinsics = torch.stack(
             [K[:, 0, 0], K[:, 1, 1], K[:, 0, 2], K[:, 1, 2]],  # fx  # fy  # cx  # cy
             dim=1,
diff --git a/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d_bilaplacian.py b/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d_bilaplacian.py
index 3f55976..7a1d0d5 100644
--- a/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d_bilaplacian.py
+++ b/roco_spring_devkit/scene_flow/models/raft_3d/raft_3d_bilaplacian.py
@@ -424,7 +424,7 @@ def predict(self, inputs, is_right_disparity=False):
         else:
             disp1_raw, disp2_raw = torch.split(inputs["disparities"], 1, dim=1)
 
-        K = inputs["intrinsics"][:, 0]
+        K = inputs["intrinsics"][:, 0, 0]
         intrinsics = torch.stack(
             [K[:, 0, 0], K[:, 1, 1], K[:, 0, 2], K[:, 1, 2]],  # fx  # fy  # cx  # cy
             dim=1,