1 files changed, 4 insertions, 2 deletions

diff --git a/util/plot_gen.py b/util/plot_gen.py
index 2ae03ef..f24fd2c 100755
--- a/util/plot_gen.py
+++ b/util/plot_gen.py
@@ -57,7 +57,8 @@ if __name__ == "__main__":
     pub_data = list(map(itemgetter(3), 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)
 
     plot_i = 1
@@ -122,4 +123,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)