diff --git a/examples/01_inference_pretrained.ipynb b/examples/01_inference_pretrained.ipynb
index bc44a86f..8dd6c8d5 100644
--- a/examples/01_inference_pretrained.ipynb
+++ b/examples/01_inference_pretrained.ipynb
@@ -21,7 +21,7 @@
    "source": [
     "from octo.model.octo_model import OctoModel\n",
     "\n",
-    "model = OctoModel.load_pretrained(\"hf://rail-berkeley/octo-small\")"
+    "model = OCTOModel.load_pretrained(\"hf://rail-berkeley/octo-small\")"
    ]
   },
   {