Fixed error in data migration