Fix axis labels in plot_joint