Fix remove.py