diff --git a/notebooks/wut-train.ipynb b/notebooks/wut-train.ipynb
index e8fb2be..739f460 100644
--- a/notebooks/wut-train.ipynb
+++ b/notebooks/wut-train.ipynb
@@ -45,20 +45,20 @@
     "import tensorflow as tf\n",
     "from tensorflow import keras\n",
     "from tensorflow.keras import layers\n",
-    "\n",
+    "from tensorflow.keras import optimizers\n",
     "from tensorflow.keras import Sequential\n",
     "from tensorflow.keras.layers import Activation, Dropout, Flatten, Dense\n",
-    "from tensorflow.keras.preprocessing.image import ImageDataGenerator\n",
     "from tensorflow.keras.layers import Convolution2D, MaxPooling2D, ZeroPadding2D\n",
-    "from tensorflow.keras import optimizers\n",
-    "from tensorflow.keras.preprocessing import image\n",
-    "from tensorflow.keras.models import load_model\n",
-    "from tensorflow.keras.preprocessing.image import load_img\n",
-    "from tensorflow.keras.preprocessing.image import img_to_array\n",
     "from tensorflow.keras.layers import Dense, Conv2D, Flatten, Dropout, MaxPooling2D\n",
-    "from tensorflow.keras.models import Model\n",
     "from tensorflow.keras.layers import Input, concatenate\n",
-    "from tensorflow.keras.utils import model_to_dot"
+    "from tensorflow.keras.models import load_model\n",
+    "from tensorflow.keras.models import Model\n",
+    "from tensorflow.keras.preprocessing import image\n",
+    "from tensorflow.keras.preprocessing.image import ImageDataGenerator\n",
+    "from tensorflow.keras.preprocessing.image import img_to_array\n",
+    "from tensorflow.keras.preprocessing.image import load_img\n",
+    "from tensorflow.keras.utils import model_to_dot\n",
+    "from tensorflow.keras.utils import plot_model"
    ]
   },
   {
@@ -300,7 +300,7 @@
    "outputs": [],
    "source": [
     "#wutoptimizer = 'adam'\n",
-    "wutoptimizer = keras.optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=True)\n",
+    "wutoptimizer = tf.keras.optimizers.Adam(learning_rate=0.001, beta_1=0.9, beta_2=0.999, amsgrad=True)\n",
     "\n",
     "wutloss = 'binary_crossentropy'\n",
     "#wutmetrics = 'accuracy'\n",
@@ -457,15 +457,6 @@
     "model.save('/srv/satnogs/data/models/GMSK/wut-GMSK-202205.tf')"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "from keras.utils import plot_model"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,