diff --git a/demo.ipynb b/demo.ipynb
index 11abb584..9c06b62c 100644
--- a/demo.ipynb
+++ b/demo.ipynb
@@ -226,7 +226,7 @@
                     "output_type": "stream",
                     "text": [
                         "\u001b[?25l     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m0.0/45.4 kB\u001b[0m \u001b[31m?\u001b[0m eta \u001b[36m-:--:--\u001b[0m\r\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m45.4/45.4 kB\u001b[0m \u001b[31m3.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
-                        "\u001b[?25h0.15.0\n"
+                        "\u001b[?25h0.16.0\n"
                     ]
                 }
             ],
@@ -540,7 +540,7 @@
                 }
             ],
             "source": [
-                "mask_annotator = sv.MaskAnnotator(color_map=\"index\")\n",
+                "mask_annotator = sv.MaskAnnotator(color_lookup=sv.ColorLookup.INDEX)\n",
                 "\n",
                 "annotated_image = mask_annotator.annotate(image.copy(), detections=detections)\n",
                 "\n",