Fix migrate.py