diff --git a/03_classification.ipynb b/03_classification.ipynb
index 5d76ee1..567f7ff 100644
--- a/03_classification.ipynb
+++ b/03_classification.ipynb
@@ -91,7 +91,7 @@
    "outputs": [],
    "source": [
     "from sklearn.datasets import fetch_openml\n",
-    "mnist = fetch_openml('mnist_784', version=1)\n",
+    "mnist = fetch_openml('mnist_784', version=1, as_frame=False)\n",
     "mnist.keys()"
    ]
   },