fix axi.py