diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 0cb574dd622e306fdb757f03cf2949d9be7bfae2..f39d303c7b53fd6ffcfc4847b13f5a84d5865ee4 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -441,7 +441,7 @@ class ESPCN:
         conv = tf.keras.layers.LeakyReLU()(conv)
         print(conv.shape)
 
-        conv = tf.keras.layers.Conv2D(num_filters/2, kernel_size=3, strides=1, padding=padding, activation=None)(conv)
+        conv = tf.keras.layers.Conv2D(num_filters/2, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
         conv = tf.keras.layers.BatchNormalization()(conv)
         print(conv.shape)