Fix migration bugs
Merge pull request #31 from taoeffect/fix-migration

Fix migration bugs