diff --git a/minerva/engines/patch_inferencer_engine.py b/minerva/engines/patch_inferencer_engine.py
index 78034a5..8fa8de3 100644
--- a/minerva/engines/patch_inferencer_engine.py
+++ b/minerva/engines/patch_inferencer_engine.py
@@ -290,7 +290,7 @@ def _compute_base_padding(self, tensor: torch.Tensor):
         Computes the padding for the base patch set based on the input tensor shape and the model's input shape.
         """
         padding = []
-        for i, t in zip(self.input_shape, tensor.shape[1:]):
+        for i, t in zip(self.padding["pad"], tensor.shape[1:]):
             padding.append(max(0, i - t))
         return padding