diff --git a/sunode/wrappers/as_pytensor.py b/sunode/wrappers/as_pytensor.py
index a9d1017..99de6e0 100644
--- a/sunode/wrappers/as_pytensor.py
+++ b/sunode/wrappers/as_pytensor.py
@@ -46,7 +46,7 @@ def read_dict(vals, name=None):
                 tensor, dim_names = vals
             else:
                 try:
-                    tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").shape.eval()
+                    tensor, dim_names = vals, pt.as_tensor_variable(vals, dtype="float64").type.shape
                 except MissingInputError as e:
                     raise ValueError(
                         'Shapes of tensors need to be statically '