diff --git a/examples/virtual_sites/vsite_showcase.ipynb b/examples/virtual_sites/vsite_showcase.ipynb
index 280902c04..b4f852397 100644
--- a/examples/virtual_sites/vsite_showcase.ipynb
+++ b/examples/virtual_sites/vsite_showcase.ipynb
@@ -436,7 +436,7 @@
     "    openmm_system: openmm.app.Simulation,\n",
     "    particle_positions: openmm.unit.Quantity,\n",
     "    minimize=False,\n",
-    ") -> tuple[numpy.ndarray, unit.Quantity]:\n",
+    ") -> tuple[numpy.ndarray, Quantity]:\n",
     "    \"\"\"\n",
     "    Calculate particle positions and potential energy of the a system.\n",
     "\n",
@@ -444,7 +444,7 @@
     "    ----------\n",
     "    openmm_topology: openmm.app.Topology, OpenMM Topology\n",
     "    openmm_system: openmm.app.System, OpenMM System\n",
-    "    particle_positions: openff.units.unit.Quantity, (N, 3) array of positions in nanometers\n",
+    "    particle_positions: openff.units.Quantity, (N, 3) array of positions in nanometers\n",
     "    minimize: bool, Whether or not to perform an energy minimization before calculating energy\n",
     "\n",
     "    Returns\n",