diff --git a/modules/image_dataset.ipynb b/modules/image_dataset.ipynb
index e431708f7..1c59091db 100644
--- a/modules/image_dataset.ipynb
+++ b/modules/image_dataset.ipynb
@@ -136,7 +136,7 @@
    ],
    "source": [
     "class TestCompose(Compose):\n",
-    "    def __call__(self, data, meta):\n",
+    "    def __call__(self, data, meta, **_kwargs):\n",
     "        data = self.transforms[0](data)  # ensure channel first\n",
     "        data = self.transforms[1](data)  # spacing\n",
     "        meta = data.meta\n",
@@ -192,7 +192,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.10"
+   "version": "3.9.16"
   }
  },
  "nbformat": 4,