diff --git a/docs/scalars_and_keras.ipynb b/docs/scalars_and_keras.ipynb
index 09afb29a74..430a9fb89f 100644
--- a/docs/scalars_and_keras.ipynb
+++ b/docs/scalars_and_keras.ipynb
@@ -204,7 +204,7 @@
         "\n",
         "model.compile(\n",
         "    loss='mse', # keras.losses.mean_squared_error\n",
-        "    optimizer=keras.optimizers.SGD(lr=0.2),\n",
+        "    optimizer=keras.optimizers.SGD(learning_rate=0.2),\n",
         ")\n",
         "\n",
         "print(\"Training ... With default parameters, this takes less than 10 seconds.\")\n",