Format pad transform

1 files changed, 1 insertions, 1 deletions

diff --git a/text_recognizer/data/transforms/pad.py b/text_recognizer/data/transforms/pad.py
index 82e4d54..1da4534 100644
--- a/text_recognizer/data/transforms/pad.py
+++ b/text_recognizer/data/transforms/pad.py
@@ -14,7 +14,7 @@ class Pad:
 
     def __call__(self, y: Tensor) -> Tensor:
         """Pads sequences with pad index if shorter than max len."""
-        if y.shape[-1] < self.length:
+        if y.shape[-1] < self.max_len:
             pad_len = self.max_len - len(y)
             y = torch.cat((y, torch.LongTensor([self.pad_index] * pad_len)))
         return y[: self.max_len]