diff --git a/cgan/cgan.py b/cgan/cgan.py
index 0cc9924d47..e66acb6e4a 100755
--- a/cgan/cgan.py
+++ b/cgan/cgan.py
@@ -35,7 +35,7 @@ def __init__(self):
 
         # The generator takes noise and the target label as input
         # and generates the corresponding digit of that label
-        noise = Input(shape=(100,))
+        noise = Input(shape=(self.latent_dim,))
         label = Input(shape=(1,))
         img = self.generator([noise, label])