Corrected wrong derivatives in axis