diff --git a/src/grid.rs b/src/grid.rs
index fb3840b..12e3f0c 100644
--- a/src/grid.rs
+++ b/src/grid.rs
@@ -140,13 +140,12 @@ where
     let bufs: Vec<_> = grids.iter().map(|grid| &grid.buf.buf).collect();
 
     // We mutate grid buffers and read grid data. We use unsafe because we need shared/unique borrows on different fields of the same Grid struct.
-    bufs.iter().enumerate().for_each(|(i, buf)| unsafe {
+    bufs.iter().enumerate().for_each(|(i, buf)| {
         let buf_ptr = *buf as *const Vec<f32> as *mut Vec<f32>;
-        buf_ptr.as_mut().unwrap().fill(0.0);
+        unsafe { buf_ptr.as_mut() }.unwrap().fill(0.0);
         datas.iter().enumerate().for_each(|(j, other)| {
             let multiplier = attraction_table[i].as_ref()[j];
-            buf_ptr
-                .as_mut()
+            unsafe { buf_ptr.as_mut() }
                 .unwrap()
                 .iter_mut()
                 .zip(*other)