diff --git a/fastmri_examples/feature_varnet/feature_varnet.py b/fastmri_examples/feature_varnet/feature_varnet.py
index 1ddb656d..7236ff61 100644
--- a/fastmri_examples/feature_varnet/feature_varnet.py
+++ b/fastmri_examples/feature_varnet/feature_varnet.py
@@ -53,9 +53,9 @@ def image_uncrop(image: Tensor, original_image: Tensor) -> Tensor:
         if len(in_shape) == 2:  # Assuming 2D images
             original_image[pad_height_top:pad_height, pad_height_left:pad_width] = image
         elif len(in_shape) == 3:  # Assuming 3D images with channels
-            original_image[:, pad_height_top:pad_height, pad_height_left:pad_width] = (
-                image
-            )
+            original_image[
+                :, pad_height_top:pad_height, pad_height_left:pad_width
+            ] = image
         elif len(in_shape) == 4:  # Assuming 4D images with batch size
             original_image[
                 :, :, pad_height_top:pad_height, pad_height_left:pad_width