diff --git a/docs/tutorials/optimizers_cyclicallearningrate.ipynb b/docs/tutorials/optimizers_cyclicallearningrate.ipynb
index a3961a7ab5..8e63a0e0fa 100644
--- a/docs/tutorials/optimizers_cyclicallearningrate.ipynb
+++ b/docs/tutorials/optimizers_cyclicallearningrate.ipynb
@@ -189,7 +189,7 @@
         "def get_training_model():\n",
         "    model = tf.keras.Sequential(\n",
         "        [\n",
-        "            layers.Input((28, 28, 1)),\n",
+        "            layers.InputLayer((28, 28, 1)),\n",
         "            layers.experimental.preprocessing.Rescaling(scale=1./255),\n",
         "            layers.Conv2D(16, (5, 5), activation=\"relu\"),\n",
         "            layers.MaxPooling2D(pool_size=(2, 2)),\n",