diff --git a/GAN/least_squares_gan/lsgan_tensorflow.py b/GAN/least_squares_gan/lsgan_tensorflow.py
index 861da28..8b9855f 100644
--- a/GAN/least_squares_gan/lsgan_tensorflow.py
+++ b/GAN/least_squares_gan/lsgan_tensorflow.py
@@ -108,7 +108,7 @@ def discriminator(x):
 
     _, G_loss_curr = sess.run(
         [G_solver, G_loss],
-        feed_dict={X: X_mb, z: sample_z(mb_size, z_dim)}
+        feed_dict={X: X_mb, z: z_mb}
     )
 
     if it % 1000 == 0: