diff --git a/v3-examples/model-customization-examples/dpo_trainer_example_notebook_v3_prod.ipynb b/v3-examples/model-customization-examples/dpo_trainer_example_notebook_v3_prod.ipynb
index 6666c75467..1965e37fb2 100644
--- a/v3-examples/model-customization-examples/dpo_trainer_example_notebook_v3_prod.ipynb
+++ b/v3-examples/model-customization-examples/dpo_trainer_example_notebook_v3_prod.ipynb
@@ -200,6 +200,8 @@
    },
    "outputs": [],
    "source": [
+    "import json\n",
+    "import re\n",
     "from pprint import pprint\n",
     "from sagemaker.core.utils.utils import Unassigned\n",
     "\n",