diff --git a/imagenet/generator.py b/imagenet/generator.py
index f4a9860..98800b8 100644
--- a/imagenet/generator.py
+++ b/imagenet/generator.py
@@ -148,7 +148,7 @@ def reuse_wrapper(packed, *args):
         sixteenth = dcgan.gf_dim // 16
         if sixteenth == 0:
             sixteenth = 1
-        h5z = make_z([dcgan.batch_size, 128, 128, eighth],
+        h5z = make_z([dcgan.batch_size, 128, 128, sixteenth],
                                    minval=-1., maxval=1.,
                                    name='h5z', dtype=tf.float32)
         zs.append(h5z)