Fix flash.py