diff --git a/docs/tutorials/losses_triplet.ipynb b/docs/tutorials/losses_triplet.ipynb
index add8e744a6..fbbd69b553 100644
--- a/docs/tutorials/losses_triplet.ipynb
+++ b/docs/tutorials/losses_triplet.ipynb
@@ -167,7 +167,7 @@
         "train_dataset = train_dataset.shuffle(1024).batch(32)\n",
         "train_dataset = train_dataset.map(_normalize_img)\n",
         "\n",
-        "test_dataset = test_dataset.shuffle(1024).batch(32)\n",
+        "test_dataset = test_dataset.batch(32)\n",
         "test_dataset = test_dataset.map(_normalize_img)"
       ],
       "execution_count": 0,