diff --git a/Donut/CORD/Fine_tune_Donut_on_a_custom_dataset_(CORD)_with_PyTorch_Lightning.ipynb b/Donut/CORD/Fine_tune_Donut_on_a_custom_dataset_(CORD)_with_PyTorch_Lightning.ipynb
index b8d892f5..12c5ec7f 100644
--- a/Donut/CORD/Fine_tune_Donut_on_a_custom_dataset_(CORD)_with_PyTorch_Lightning.ipynb
+++ b/Donut/CORD/Fine_tune_Donut_on_a_custom_dataset_(CORD)_with_PyTorch_Lightning.ipynb
@@ -8233,7 +8233,7 @@
         "output_list = []\n",
         "accs = []\n",
         "\n",
-        "dataset = load_dataset(\"naver-clova-ix/cord-v2\", split=\"validation\")\n",
+        "dataset = load_dataset(\"naver-clova-ix/cord-v2\", split=\"test\")\n",
         "\n",
         "for idx, sample in tqdm(enumerate(dataset), total=len(dataset)):\n",
         "    # prepare encoder inputs\n",
@@ -28079,4 +28079,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}