diff --git a/video_chat2/mvbench.ipynb b/video_chat2/mvbench.ipynb
index 7886b79..802d26c 100644
--- a/video_chat2/mvbench.ipynb
+++ b/video_chat2/mvbench.ipynb
@@ -31,7 +31,7 @@
     "import io\n",
     "import json\n",
     "\n",
-    "from models.videochat2_it import VideoChat2_it\n",
+    "from models.videochat_vicuna.videochat2_it_vicuna import VideoChat2_it_vicuna as VideoChat2_it\n",
     "from utils.easydict import EasyDict\n",
     "import torch\n",
     "\n",