fixed axi bug