diff --git a/gstreamer-analytics/src/tensor.rs b/gstreamer-analytics/src/tensor.rs
index c52a00d90..4403b575b 100644
--- a/gstreamer-analytics/src/tensor.rs
+++ b/gstreamer-analytics/src/tensor.rs
@@ -84,18 +84,17 @@ impl Tensor {
     #[doc(alias = "gst_tensor_check_type")]
     pub fn check_type(
         &self,
-        order: crate::TensorDimOrder,
-        num_dims: usize,
         data_type: crate::TensorDataType,
-        data: &gst::BufferRef,
+        order: crate::TensorDimOrder,
+        dims: &[usize],
     ) -> bool {
         unsafe {
             from_glib(ffi::gst_tensor_check_type(
                 self.to_glib_none().0,
-                order.into_glib(),
-                num_dims,
                 data_type.into_glib(),
-                mut_override(data.as_ptr()),
+                order.into_glib(),
+                dims.len(),
+                dims.as_ptr(),
             ))
         }
     }
diff --git a/gstreamer-analytics/src/tensor_meta.rs b/gstreamer-analytics/src/tensor_meta.rs
index 07e2ecf26..0137a18da 100644
--- a/gstreamer-analytics/src/tensor_meta.rs
+++ b/gstreamer-analytics/src/tensor_meta.rs
@@ -54,19 +54,18 @@ impl TensorMeta {
     pub fn typed_tensor(
         &self,
         id: glib::Quark,
-        order: crate::TensorDimOrder,
-        num_dims: usize,
         data_type: crate::TensorDataType,
-        data: &gst::BufferRef,
+        order: crate::TensorDimOrder,
+        dims: &[usize],
     ) -> Option<&crate::Tensor> {
         unsafe {
             let res = ffi::gst_tensor_meta_get_typed_tensor(
                 self.as_mut_ptr(),
                 id.into_glib(),
-                order.into_glib(),
-                num_dims,
                 data_type.into_glib(),
-                mut_override(data.as_ptr()),
+                order.into_glib(),
+                dims.len(),
+                dims.as_ptr(),
             );
             if res.is_null() {
                 None