Fix import script