diff --git a/tensorflow_classification/Test5_resnet/predict.py b/tensorflow_classification/Test5_resnet/predict.py
index a2e35313f..60cf035f4 100644
--- a/tensorflow_classification/Test5_resnet/predict.py
+++ b/tensorflow_classification/Test5_resnet/predict.py
@@ -37,7 +37,7 @@
 model = tf.keras.Sequential([feature,
                              tf.keras.layers.GlobalAvgPool2D(),
                              tf.keras.layers.Dropout(rate=0.5),
-                             tf.keras.layers.Dense(1024),
+                             tf.keras.layers.Dense(1024, activation="relu"),
                              tf.keras.layers.Dropout(rate=0.5),
                              tf.keras.layers.Dense(5),
                              tf.keras.layers.Softmax()])