diff --git a/Inst_eddies/Analysis/plot_velocity.py b/Inst_eddies/Analysis/plot_velocity.py
index e4b77c918b8d8e5b71ac3eed43f5f16b834ef27f..4a3dd3fe48cc020553820762e84e98838e4f24ca 100755
--- a/Inst_eddies/Analysis/plot_velocity.py
+++ b/Inst_eddies/Analysis/plot_velocity.py
@@ -75,14 +75,14 @@ with netCDF4.Dataset(args.input_file) as f:
                                   vgos[lat_mask][:, lon_mask],
                                   scale = args.scale, scale_units = "width",
                                   transform = src_crs)
-        plt.quiverkey(quiver_return, 0.9, 0.9, 1, r"1 m s$^{-1}$",
-                      coordinates = "figure")
+        ax.quiverkey(quiver_return, 0.9, 0.9, 1, r"1 m s$^{-1}$",
+                     coordinates = "figure")
 
 ax.gridlines(draw_labels = True)
 ax.coastlines()
 
 if args.save:
-    plt.savefig(f"plot_velocity.{args.save}")
+    fig.savefig(f"plot_velocity.{args.save}")
     print(f'Created "plot_velocity.{args.save}".')
 else:
     plt.show()