fix A.py