Revert "Fix migration"

This reverts commit 6e38eea77eb7a56afecd46c7d1ac2b506f34b629.
Merge pull request #65 from gary-pei/master

Revert "Fix migration"