Fix AX submodule

AX submodule