flash: add missing changes for flash2 support