diff --git a/src/meta_sl/load_dataset/load_sinusoid.ipynb b/src/meta_sl/load_dataset/load_sinusoid.ipynb
index dc24007..1479d1e 100644
--- a/src/meta_sl/load_dataset/load_sinusoid.ipynb
+++ b/src/meta_sl/load_dataset/load_sinusoid.ipynb
@@ -65,7 +65,7 @@
    "source": [
     "config = {\n",
     "    \"num_shots\": 5,\n",
-    "    \"batch_size\": 3,\n",
+    "    \"batch_size\": 16,\n",
     "    \"num_batches_train\": 6000,\n",
     "    \"num_batches_test\": 2000,\n",
     "    \"num_batches_val\": 100,\n",