Fix finding of the axis