diff --git a/Makefile b/Makefile
index 7479b9c..eed7da3 100644
--- a/Makefile
+++ b/Makefile
@@ -68,3 +68,4 @@ install: clean install-heat
 	chmod +x analysis/start_01_grid.sh
 	chmod +x analysis/start_02_nl2sol.sh
 	chmod +x analysis/start_03_ego.sh
+	python figures/make_black_box_plot.py
diff --git a/figures/black_box.png b/figures/black_box.png
deleted file mode 100644
index b4ec501..0000000
Binary files a/figures/black_box.png and /dev/null differ
diff --git a/figures/make_black_box_plot.py b/figures/make_black_box_plot.py
index b7bd73e..3fb83b1 100644
--- a/figures/make_black_box_plot.py
+++ b/figures/make_black_box_plot.py
@@ -1,3 +1,5 @@
+import os
+
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
 
@@ -53,4 +55,4 @@
 
 ax.axis([0, 30, 0, 7])
 ax.axis("off")
-plt.savefig("black_box.png")
+plt.savefig(os.path.join("figures", "black_box.png"))