diff --git a/openfl-tutorials/experimental/Workflow_Interface_104_Keras_MNIST_with_GPU.ipynb b/openfl-tutorials/experimental/Workflow_Interface_104_Keras_MNIST_with_GPU.ipynb
index ab46915c10e..0df9bf8db2b 100644
--- a/openfl-tutorials/experimental/Workflow_Interface_104_Keras_MNIST_with_GPU.ipynb
+++ b/openfl-tutorials/experimental/Workflow_Interface_104_Keras_MNIST_with_GPU.ipynb
@@ -77,10 +77,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from keras.layers import Flatten, Dense, Dropout, Conv2D, MaxPool2D\n",
-    "from keras.models import Sequential\n",
-    "from keras.datasets import mnist\n",
-    "from keras.utils import np_utils\n",
+    "from tensorflow.keras.layers import Flatten, Dense, Dropout, Conv2D, MaxPool2D\n",
+    "from tensorflow.keras.models import Sequential\n",
+    "from tensorflow.keras.datasets import mnist\n",
+    "from tensorflow.keras.utils import to_categorical\n",
     "\n",
     "nb_classes = 10\n",
     "(X_train, y_train), (X_test, y_test) = mnist.load_data()\n",
@@ -94,8 +94,8 @@
     "print(\"Training matrix shape\", X_train.shape)\n",
     "print(\"Testing matrix shape\", X_test.shape)\n",
     "\n",
-    "Y_train = np_utils.to_categorical(y_train, nb_classes)\n",
-    "Y_test = np_utils.to_categorical(y_test, nb_classes)\n",
+    "Y_train = to_categorical(y_train, nb_classes)\n",
+    "Y_test = to_categorical(y_test, nb_classes)\n",
     "\n",
     "\n",
     "model = Sequential([\n",