Fix demo import script