Fixed _finalize_axis for plotting