Fix logic in import script.