Revert "Fixed migration bug"

This reverts commit 948a846081063b4c8a1dfb786c9cdb2fbf5e6c62.