AxeXpLoss fixed

Other tools still to fix