Simplify compute_axis