fix AB.py