Fixed some migration