Fixed _finalize_axis method