Revert "Fix migration"

This reverts commit 6e38eea77eb7a56afecd46c7d1ac2b506f34b629.