diff --git a/DeepSpeech.ipynb b/DeepSpeech.ipynb
index e9752234..9fa4cdb5 100644
--- a/DeepSpeech.ipynb
+++ b/DeepSpeech.ipynb
@@ -607,7 +607,7 @@
    "source": [
     "def calculate_accuracy_and_loss(batch_set):\n",
     "    # Obtain the next batch of data\n",
-    "    batch_x, batch_y, n_steps = ted_lium.train.next_batch()\n",
+    "    batch_x, batch_y, n_steps = batch_set.next_batch()\n",
     "\n",
     "    # Set batch_seq_len for the batch\n",
     "    batch_seq_len = batch_x.shape[0] * [n_steps]\n",
@@ -645,7 +645,7 @@
     "```python\n",
     "def calculate_accuracy_and_loss(batch_set):\n",
     "    # Obtain the next batch of data\n",
-    "    batch_x, batch_y, n_steps = ted_lium.train.next_batch()\n",
+    "    batch_x, batch_y, n_steps = batch_set.next_batch()\n",
     "```\n",
     "simply obtian the next mini-batch of data.\n",
     "\n",