diff --git a/+sleap/+model/+topdown/+internal/Postprocess1CustomLayer.m b/+sleap/+model/+topdown/+internal/Postprocess1CustomLayer.m
index 9386eb5..e509d16 100644
--- a/+sleap/+model/+topdown/+internal/Postprocess1CustomLayer.m
+++ b/+sleap/+model/+topdown/+internal/Postprocess1CustomLayer.m
@@ -182,6 +182,17 @@
             crops = zeros(layer.BboxSize(1), layer.BboxSize(2), 1, size(Z1, 4), 'like',inputIm);
 
             n=1;
+
+            % % Reshape peaks for indexing
+            % x = reshape(Z1(1, 1, 1, 1:n), [1, 1, 1, n]);
+            % y = reshape(Z1(1, 1, 2, 1:n), [1, 1, 1, n]);
+            % 
+            % % Create reference object for cropping
+            % R = imref2d(layer.BboxSize, [x - layer.BboxSize(1)/2, x + layer.BboxSize(1)/2], [y - layer.BboxSize(2)/2, y + layer.BboxSize(2)/2]);
+            % 
+            % % Perform cropping using indexing
+            % crops(:,:,1,1:n) = imwarp(inputIm, affine2d(), 'OutputView', R);
+            
             for i = 1:size(inputIm,4)
                 for j = 1:numAnimalsPerImage(1,1,1,i)
                     x = Z1(1,1,1,n);
@@ -228,4 +239,4 @@
         end
 
     end
-end
\ No newline at end of file
+end
diff --git a/demo.mlx b/demo.mlx
index 118332c..0085b9d 100644
Binary files a/demo.mlx and b/demo.mlx differ