diff --git a/site/en/guide/intro_to_modules.ipynb b/site/en/guide/intro_to_modules.ipynb
index 1bedb87e894..ef8311e44d5 100644
--- a/site/en/guide/intro_to_modules.ipynb
+++ b/site/en/guide/intro_to_modules.ipynb
@@ -918,6 +918,8 @@
       },
       "outputs": [],
       "source": [
+        "import keras
+        "@keras.saving.register_keras_serializable()
         "class MySequentialModel(tf.keras.Model):\n",
         "  def __init__(self, name=None, **kwargs):\n",
         "    super().__init__(**kwargs)\n",