diff --git a/labs/OATD_LR4_metod.ipynb b/labs/OATD_LR4_metod.ipynb
index 7d7f1a0..128279a 100644
--- a/labs/OATD_LR4_metod.ipynb
+++ b/labs/OATD_LR4_metod.ipynb
@@ -918,7 +918,7 @@
     "    if states is None:\n",
     "        states = self.gru.get_initial_state(x)\n",
     "\n",
-    "    x, states = self.gru(x, initial_state=states, training=training)\n",
+    "    x, *states = self.gru(x, initial_state=states, training=training)\n",
     "    x = self.dense(x, training=training)\n",
     "\n",
     "    if return_state:\n",