Fix axis label class