diff --git a/Tutorial14/Tutorial14.ipynb b/Tutorial14/Tutorial14.ipynb
index 0698219..5ffd764 100644
--- a/Tutorial14/Tutorial14.ipynb
+++ b/Tutorial14/Tutorial14.ipynb
@@ -332,7 +332,7 @@
    "source": [
     "name = 'Cora'\n",
     "transform = transforms.Compose([\n",
-    "    transforms.AddTrainValTestMask('train_rest', num_val=500, num_test=500),\n",
+    "    transforms.RandomNodeSplit('train_rest', num_val=500, num_test=500),\n",
     "    transforms.TargetIndegree(),\n",
     "])\n",
     "cora = datasets.Planetoid('./data', name, pre_transform=transforms.NormalizeFeatures(), transform=transform)"