Fix data import script