diff --git a/generate.py b/generate.py
index ecfd2bf6d..d186df7eb 100644
--- a/generate.py
+++ b/generate.py
@@ -264,7 +264,7 @@ def main():
     tf.summary.audio('generated', decode, wavenet_params['sample_rate'])
     summaries = tf.summary.merge_all()
     summary_out = sess.run(summaries,
-                           feed_dict={samples: np.reshape(waveform, [-1, 1])})
+                           feed_dict={samples: np.reshape(waveform, [1, -1])})
     writer.add_summary(summary_out)
 
     # Save the result as a wav file.