fixed typo
added missing migration arg
Merge pull request #253 from yuyiguo/migration-fix

Migration fix