diff --git a/modules/deeplearning/srcnn_cld_frac.py b/modules/deeplearning/srcnn_cld_frac.py
index 4819d9387923fecbff28dbfa00733f8a55d96975..83fea1498c8484046015ddf4490b467508fcc2a4 100644
--- a/modules/deeplearning/srcnn_cld_frac.py
+++ b/modules/deeplearning/srcnn_cld_frac.py
@@ -528,7 +528,7 @@ class SRCNN:
             print(conv.shape)
             conv = tf.nn.depth_to_space(conv, factor)
             print(conv.shape)
-            self.logits = tf.keras.layers.Conv2D(IMG_DEPTH, kernel_size=3, strides=1, padding=padding, name='regression')(conv)
+            self.logits = tf.keras.layers.Conv2D(IMG_DEPTH, kernel_size=3, strides=1, padding=padding, activation=final_activation)(conv)
         print(self.logits.shape)
 
     def build_training(self):