diff --git a/advanced_tutorials/recommender-system/2_train_retrieval_model.ipynb b/advanced_tutorials/recommender-system/2_train_retrieval_model.ipynb
index 48b1bb90..98cf58b0 100644
--- a/advanced_tutorials/recommender-system/2_train_retrieval_model.ipynb
+++ b/advanced_tutorials/recommender-system/2_train_retrieval_model.ipynb
@@ -27,9 +27,9 @@
    "outputs": [],
    "source": [
     "import tensorflow as tf\n",
-    "from tensorflow.keras.layers.experimental.preprocessing import StringLookup, Normalization\n",
+    "from tensorflow.keras.layers import StringLookup, Normalization\n",
+    "\n",
     "import tensorflow_recommenders as tfrs\n",
-    "import tensorflow_addons as tfa\n",
     "\n",
     "import warnings\n",
     "warnings.filterwarnings('ignore')"
@@ -482,7 +482,7 @@
     "model = TwoTowerModel(query_model, item_model)\n",
     "\n",
     "# Define an optimizer using AdamW with a learning rate of 0.01\n",
-    "optimizer = tfa.optimizers.AdamW(0.001, learning_rate=0.01)\n",
+    "optimizer = tf.keras.optimizers.AdamW(weight_decay=0.001, learning_rate=0.01)\n",
     "\n",
     "# Compile the model using the specified optimizer\n",
     "model.compile(optimizer=optimizer)"
@@ -710,7 +710,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "Python 3",
    "language": "python",
    "name": "python3"
   },
@@ -724,7 +724,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.18"
+   "version": "3.10.11"
   }
  },
  "nbformat": 4,
diff --git a/advanced_tutorials/recommender-system/requirements.txt b/advanced_tutorials/recommender-system/requirements.txt
index 3371e075..53888762 100644
--- a/advanced_tutorials/recommender-system/requirements.txt
+++ b/advanced_tutorials/recommender-system/requirements.txt
@@ -1,8 +1,7 @@
 #These are the libraries you need to install to the Hopsworks cluster.
-tensorflow==2.11
+tensorflow==2.13
 tensorflow-recommenders-0.7.2
-tensorflow-addons-0.22.0
 catboost==1.1.1
 opensearch-py==1.1.0
 hopsworks
-streamlit==1.28.2
\ No newline at end of file
+streamlit==1.28.2