fix prep.py