diff --git a/codes/models/archs/discriminator_vgg_arch.py b/codes/models/archs/discriminator_vgg_arch.py
index e0f7ca78..0a109247 100644
--- a/codes/models/archs/discriminator_vgg_arch.py
+++ b/codes/models/archs/discriminator_vgg_arch.py
@@ -177,7 +177,7 @@ class Discriminator_UNet(nn.Module):
     def __init__(self, in_nc, nf):
         super(Discriminator_UNet, self).__init__()
         # [64, 128, 128]
-        self.conv0_0 = ConvGnLelu(in_nc, nf, kernel_size=3, bias=True, gn=False)
+        self.conv0_0 = ConvGnLelu(in_nc, nf, kernel_size=3, bias=True, activation=False)
         self.conv0_1 = ConvGnLelu(nf, nf, kernel_size=3, stride=2, bias=False)
         # [64, 64, 64]
         self.conv1_0 = ConvGnLelu(nf, nf * 2, kernel_size=3, bias=False)