diff --git a/ml/pc/exercises/image_classification_part1.ipynb b/ml/pc/exercises/image_classification_part1.ipynb
index 369acd4..e852523 100644
--- a/ml/pc/exercises/image_classification_part1.ipynb
+++ b/ml/pc/exercises/image_classification_part1.ipynb
@@ -438,7 +438,7 @@
         "from tensorflow.keras.optimizers import RMSprop\n",
         "\n",
         "model.compile(loss='binary_crossentropy',\n",
-        "              optimizer=RMSprop(lr=0.001),\n",
+        "              optimizer=RMSprop(learning_rate=0.001),\n",
         "              metrics=['acc'])"
       ]
     },