added support for new migrate feature