fix some migration
Merge pull request #50 from blackbiron/master

fix some migration