Fix bug in clean.py