diff --git a/lab1/Part1_TensorFlow.ipynb b/lab1/Part1_TensorFlow.ipynb
index 25e4894e..8b6f611a 100644
--- a/lab1/Part1_TensorFlow.ipynb
+++ b/lab1/Part1_TensorFlow.ipynb
@@ -352,7 +352,7 @@
         "    return y\n",
         "\n",
         "# Since layer parameters are initialized randomly, we will set a random seed for reproducibility\n",
-        "tf.keras.utils.set_random_seed(1)\n",
+        "tf.random.set_seed(1)\n",
         "layer = OurDenseLayer(3)\n",
         "layer.build((1,2))\n",
         "x_input = tf.constant([[1,2.]], shape=(1,2))\n",
@@ -695,4 +695,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}