diff --git a/lightweight_gan/lightweight_gan.py b/lightweight_gan/lightweight_gan.py
index b3a4c7b..d751347 100644
--- a/lightweight_gan/lightweight_gan.py
+++ b/lightweight_gan/lightweight_gan.py
@@ -567,7 +567,7 @@ def __init__(
         self.residual_layers = nn.ModuleList([])
 
         for (res, ((_, chan_in), (_, chan_out))) in zip(non_residual_resolutions, chan_in_out):
-            image_width = 2 ** resolution
+            image_width = 2 ** res
 
             attn = None
             if image_width in attn_res_layers:
diff --git a/lightweight_gan/version.py b/lightweight_gan/version.py
index f7d1e3e..5ec52a9 100644
--- a/lightweight_gan/version.py
+++ b/lightweight_gan/version.py
@@ -1 +1 @@
-__version__ = '0.17.7'
+__version__ = '0.18.0'