diff --git a/image_ref/main.py b/image_ref/main.py
index 1d73a0d068424f7fea748d2b1d4144a2cd651d86..9ac992811634d342d5047a61d2cfa3ba9e6bdac7 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -120,6 +120,7 @@ def run_duo(args):
     plt.ylabel('accuracy')
     plt.legend(loc="upper left")
     plt.ylim(0, 1.05)
+    plt.tight_layout()
 
     plt.subplot(2, 1, 2)
     plt.plot(train_loss, label='train')
@@ -128,6 +129,7 @@ def run_duo(args):
     plt.xlabel('epoch')
     plt.ylabel('loss')
     plt.legend(loc="upper left")
+    plt.tight_layout()
 
 
     plt.show()