diff --git a/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb b/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb
index 594164a..e74045f 100644
--- a/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb
+++ b/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb
@@ -315,7 +315,7 @@
     "        layers.Flatten(input_shape=(32,32,3)),\n",
     "        layers.Dense(3000, activation='relu'),\n",
     "        layers.Dense(1000, activation='relu'),\n",
-    "        layers.Dense(10, activation='sigmoid')    \n",
+    "        layers.Dense(10, activation='softmax')    \n",
     "    ])\n",
     "\n",
     "ann.compile(optimizer='SGD',\n",
@@ -664,9 +664,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.7.3"
   }
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
\ No newline at end of file
+}