Fix data migration