diff --git a/demo/wp_scalogram.py b/demo/wp_scalogram.py
index 5df9afed..31be199a 100644
--- a/demo/wp_scalogram.py
+++ b/demo/wp_scalogram.py
@@ -38,12 +38,13 @@
 # Show spectrogram and wavelet packet coefficients
 fig2 = plt.figure()
 ax2 = fig2.add_subplot(211)
-ax2.specgram(data, NFFT=64, noverlap=32, Fs=2, cmap=cmap,
+ax2.specgram(data, NFFT=64, noverlap=32, Fs=512, cmap=cmap,
              interpolation='bilinear')
 ax2.set_title("Spectrogram of signal")
 ax3 = fig2.add_subplot(212)
-ax3.imshow(values, origin='upper', extent=[-1, 1, -1, 1],
-           interpolation='nearest')
+ax3.imshow(values, origin='lower', extent=[0, 1, 0,  len(values)], 
+           interpolation='nearest', aspect='auto')
+ax3.set_yticks(np.arange(0.5, len(labels) + 0.5), labels)
 ax3.set_title("Wavelet packet coefficients")