diff --git a/6 - Transformers for Sentiment Analysis.ipynb b/6 - Transformers for Sentiment Analysis.ipynb
index 36934cd..03e22db 100644
--- a/6 - Transformers for Sentiment Analysis.ipynb	
+++ b/6 - Transformers for Sentiment Analysis.ipynb	
@@ -553,7 +553,7 @@
     "        \n",
     "        _, hidden = self.rnn(embedded)\n",
     "        \n",
-    "        #hidden = [n layers * n directions, batch size, emb dim]\n",
+    "        #hidden = [n layers * n directions, batch size, hid dim]\n",
     "        \n",
     "        if self.rnn.bidirectional:\n",
     "            hidden = self.dropout(torch.cat((hidden[-2,:,:], hidden[-1,:,:]), dim = 1))\n",
@@ -1042,4 +1042,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
\ No newline at end of file
+}