fix swap.py