Fix up import script