Fix migration
Merge pull request #272 from kawazrepos/fix_migration

Fix migration