Custom migration-to be revert