diff --git a/examples/virtual_staining/vcp_tutorials/quick_start.ipynb b/examples/virtual_staining/vcp_tutorials/quick_start.ipynb
index bd991b28..cf01087a 100644
--- a/examples/virtual_staining/vcp_tutorials/quick_start.ipynb
+++ b/examples/virtual_staining/vcp_tutorials/quick_start.ipynb
@@ -282,6 +282,8 @@
     "# VSCyto2D is fine-tuned from a FCMAE-pretrained UNeXt2 model.\n",
     "# See this module for options to configure the model:\n",
     "\n",
+    "from viscy.unet.networks.fcmae import FullyConvolutionalMAE\n",
+    "\n",
     "?FullyConvolutionalMAE"
    ]
   },