diff --git a/train.py b/train.py
index 08a1f5a2..fcd4d92f 100644
--- a/train.py
+++ b/train.py
@@ -106,7 +106,7 @@ def train(model, criterion, data_loader, optimizer, epoch):
             
         # create attention mask
         # TODO: vectorize
-        print(tet_input_var.shape)
+        print(text_input_var.shape)
         print(mel_spec_var.shape)
         N = text_input_var.shape[1]
         T = mel_spec_var.shape[1]