diff --git a/models/networks/normalization.py b/models/networks/normalization.py
index f4977b66..d5c2a59c 100644
--- a/models/networks/normalization.py
+++ b/models/networks/normalization.py
@@ -26,6 +26,8 @@ def add_norm_layer(layer):
         if norm_type.startswith('spectral'):
             layer = spectral_norm(layer)
             subnorm_type = norm_type[len('spectral'):]
+        else:
+            subnorm_type = norm_type
 
         if subnorm_type == 'none' or len(subnorm_type) == 0:
             return layer