diff --git a/embodichain/lab/sim/solvers/srs_solver.py b/embodichain/lab/sim/solvers/srs_solver.py
index 8558ea2..796e659 100644
--- a/embodichain/lab/sim/solvers/srs_solver.py
+++ b/embodichain/lab/sim/solvers/srs_solver.py
@@ -568,7 +568,7 @@ def get_ik(
             return (
                 torch.zeros(num_targets, dtype=torch.bool, device=self.device),
                 torch.zeros(
-                    (num_targets, num_targets, 7),
+                    (num_targets, 7),
                     dtype=qpos_seed.dtype,
                     device=self.device,
                 ),