1 files changed, 4 insertions, 2 deletions

diff --git a/util/plot_dh.py b/util/plot_dh.py
index 327e28a..55e11b3 100755
--- a/util/plot_dh.py
+++ b/util/plot_dh.py
@@ -47,7 +47,8 @@ if __name__ == "__main__":
     secret_data = list(map(itemgetter(4), data))
 
     plt.style.use("ggplot")
-    fig = plt.figure(tight_layout=True)
+    fig = plt.figure()
+    fig.tight_layout(rect=[0, 0.02, 1, 0.98])
     fig.suptitle(opts.file)
 
     axe_hist = fig.add_subplot(2,1,1)
@@ -95,4 +96,5 @@ if __name__ == "__main__":
         plt.show()
     else:
         fig.set_size_inches(12, 10)
-        plt.savefig(opts.output, dpi=400)
+        ext = opts.output.name.split(".")[-1]
+        plt.savefig(opts.output, format=ext, dpi=400)