diff --git a/examples/model_diagnostics/custom_metrics_basics.ipynb b/examples/model_diagnostics/custom_metrics_basics.ipynb
index 2d282f8ec..45c3a8c4b 100644
--- a/examples/model_diagnostics/custom_metrics_basics.ipynb
+++ b/examples/model_diagnostics/custom_metrics_basics.ipynb
@@ -106,7 +106,8 @@
                 "from labelbox.data.annotation_types import (ScalarMetric, Label, ImageData,\n",
                 "                                            Point, Rectangle, ObjectAnnotation,\n",
                 "                                            ClassificationAnnotation,\n",
-                "                                            ClassificationAnswer, Radio)"
+                "                                            ClassificationAnswer, Radio)",
+                "import json"
             ]
         },
         {