diff --git a/README.md b/README.md
index 02320d4bc..178189c43 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 [![ci][19]][16] [![docs][18]][3] [![book][22]][21] [![slack][17]][4] [![github-discussions][20]][5]
 
-# Arrayfire Rust Bindings
+# ArrayFire Rust Bindings
 
 [ArrayFire][1] is a high performance library for parallel computing with an easy-to-use API. It
 enables users to write scientific computing code that is portable across CUDA, OpenCL and CPU
diff --git a/src/ml/mod.rs b/src/ml/mod.rs
index 1080a4768..e448e023f 100644
--- a/src/ml/mod.rs
+++ b/src/ml/mod.rs
@@ -76,11 +76,11 @@ where
             &mut temp as *mut af_array,
             signal.get(),
             filter.get(),
-            strides.ndims() as c_uint,
+            4,
             strides.get().as_ptr() as *const dim_t,
-            padding.ndims() as c_uint,
+            4,
             padding.get().as_ptr() as *const dim_t,
-            dilation.ndims() as c_uint,
+            4,
             dilation.get().as_ptr() as *const dim_t,
         );
         HANDLE_ERROR(AfError::from(err_val));
@@ -126,11 +126,11 @@ where
             original_signal.get(),
             original_filter.get(),
             convolved_output.get(),
-            strides.ndims() as c_uint,
+            4,
             strides.get().as_ptr() as *const dim_t,
-            padding.ndims() as c_uint,
+            4,
             padding.get().as_ptr() as *const dim_t,
-            dilation.ndims() as c_uint,
+            4,
             dilation.get().as_ptr() as *const dim_t,
             grad_type as c_uint,
         );