diff --git a/src/Restorer/Cond_NAF.py b/src/Restorer/Cond_NAF.py
index 5dc2d0f..1deaa46 100644
--- a/src/Restorer/Cond_NAF.py
+++ b/src/Restorer/Cond_NAF.py
@@ -281,7 +281,7 @@ def forward(self, input):
 
         x = self.dropout2(x)
 
-        x = self.sca_out(x) * x
+        x = self.sca_out(x, cond) * x
 
 
         return (y + x * self.gamma, cond)