diff --git a/notebooks/wut-train.ipynb b/notebooks/wut-train.ipynb
index 61270d3..57916f9 100644
--- a/notebooks/wut-train.ipynb
+++ b/notebooks/wut-train.ipynb
@@ -105,7 +105,6 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "train_dir = os.path.join('/srv/satnogs/data/txmodes', ENCODING )\n",
     "train_dir = os.path.join('/srv/satnogs/data/txmodes', ENCODING, 'train')\n",
     "val_dir = os.path.join('/srv/satnogs/data/txmodes', ENCODING, 'val')\n",
     "train_good_dir = os.path.join(train_dir, 'good')\n",
@@ -130,17 +129,17 @@
     "print('total training bad images:', num_train_bad)\n",
     "#print(\"--\")\n",
     "print(\"Total training images:\", total_train)\n",
-    "#print('total validation good images:', num_val_good)\n",
-    "#print('total validation bad images:', num_val_bad)\n",
-    "#print(\"--\")\n",
-    "#print(\"Total validation images:\", total_val)\n",
+    "print('total validation good images:', num_val_good)\n",
+    "print('total validation bad images:', num_val_bad)\n",
+    "print(\"--\")\n",
+    "print(\"Total validation images:\", total_val)\n",
     "#print(\"Reduce training and validation set when testing\")\n",
-    "total_train = 100\n",
-    "total_val = 100\n",
-    "#print(\"Train =\")\n",
-    "#print(total_train)\n",
-    "#print(\"Validation =\")\n",
-    "#print(total_val)"
+    "#total_train = 100\n",
+    "#total_val = 100\n",
+    "print(\"Train =\")\n",
+    "print(total_train)\n",
+    "print(\"Validation =\")\n",
+    "print(total_val)"
    ]
   },
   {