diff --git a/docs/wandb.ipynb b/docs/wandb.ipynb
index bead060..ed0daeb 100644
--- a/docs/wandb.ipynb
+++ b/docs/wandb.ipynb
@@ -175,7 +175,7 @@
    },
    "outputs": [],
    "source": [
-    "training_data_artifact = ln.Artifact.filter(description=\"MNIST-dataset\").one()\n",
+    "training_data_artifact = ln.Artifact.get(description=\"MNIST-dataset\")\n",
     "training_data_artifact"
    ]
   },