diff --git a/speed_test/plot_speed_test_results.py b/speed_test/plot_speed_test_results.py
index 326fa15..0b52e15 100644
--- a/speed_test/plot_speed_test_results.py
+++ b/speed_test/plot_speed_test_results.py
@@ -30,6 +30,7 @@
     axs[0].plot(
         insizes,
         p / n,
+        label=f"numba with output size: {size}",
         color=red_gradient[j],
     )
 
@@ -53,7 +54,6 @@
     axs[1].plot(
         insizes,
         p_e / n_e,
-        label=f"numba with output size: {size}",
         color=red_gradient[k],
     )
 
@@ -74,7 +74,6 @@
 axs[1].grid()
 
 axs[0].legend()
-axs[1].legend()
 
 axs[0].xaxis.set_major_formatter(ScalarFormatter(useMathText=False))
 axs[1].xaxis.set_major_formatter(ScalarFormatter(useMathText=False))
@@ -90,8 +89,8 @@
 axs[0].set_xlim(150, 4250)
 axs[1].set_xlim(150, 4250)
 
-axs[0].set_ylim(0, 250)
-axs[1].set_ylim(0, 250)
+axs[0].set_ylim(0, 400)
+axs[1].set_ylim(0, 400)
 
 axs[1].set_yticklabels([])
 
diff --git a/speed_test/speed_test.png b/speed_test/speed_test.png
index 640b0a4..b7c6d41 100644
Binary files a/speed_test/speed_test.png and b/speed_test/speed_test.png differ