diff --git a/tools/plot_builtin_wave.py b/tools/plot_builtin_wave.py
index 8ec95172..bd5fb42b 100644
--- a/tools/plot_builtin_wave.py
+++ b/tools/plot_builtin_wave.py
@@ -124,8 +124,10 @@ else:
     ax1.set_ylabel('Amplitude')
 
 [ax.grid() for ax in fig.axes] # Turn on grid
-plt.show()
 
-# Save a PDF of the figure
+# Save a PDF/PNG of the figure
+#fig.savefig(os.path.dirname(os.path.abspath(__file__)) + os.sep + w.type + '.pdf', dpi=None, format='pdf', bbox_inches='tight', pad_inches=0.1)
 #fig.savefig(os.path.dirname(os.path.abspath(__file__)) + os.sep + w.type + '.png', dpi=150, format='png', bbox_inches='tight', pad_inches=0.1)
 
+plt.show()
+