diff --git a/demo.ipynb b/demo.ipynb
index c4e50480..78518cb9 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -382,11 +382,11 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "71ab5306",
+   "id": "603c5533-f7b5-4e2c-ae62-de047abce7cc",
    "metadata": {},
    "outputs": [],
    "source": [
-    "pose = pycolmap.Image(tvec=ret[\"tvec\"], qvec=ret[\"qvec\"])\n",
+    "pose = pycolmap.Image(cam_from_world=ret[\"cam_from_world\"])\n",
     "viz_3d.plot_camera_colmap(\n",
     "    fig, pose, camera, color=\"rgba(0,255,0,0.5)\", name=query, fill=True\n",
     ")\n",