Fixed some migration script