diff --git a/tasks/human_pose/live_demo.ipynb b/tasks/human_pose/live_demo.ipynb
index 9406412..8c8c534 100644
--- a/tasks/human_pose/live_demo.ipynb
+++ b/tasks/human_pose/live_demo.ipynb
@@ -14,7 +14,7 @@
    "outputs": [],
    "source": [
     "import json\n",
-    "import trt_pose.coco\n",
+    "import trt_pose\n",
     "\n",
     "with open('human_pose.json', 'r') as f:\n",
     "    human_pose = json.load(f)\n",