diff --git a/models/tokenizer.py b/models/tokenizer.py
index c9310f9b..049e7d1b 100644
--- a/models/tokenizer.py
+++ b/models/tokenizer.py
@@ -63,7 +63,7 @@ def vq_encode(self, videos: Any, training: bool = True) -> Dict[str, Any]:
         indices = indices.reshape(B, T, N)
         return dict(z_q=z_q, z=z, emb=emb, indices=indices)
 
-    def decode(self, indices: Any, video_hw: Tuple[int]):
+    def decode(self, indices: Any, video_hw: Tuple[int, int]):
         z = self.vq.codebook[indices]
         recon = self.decoder(z)
         recon = nn.sigmoid(recon)