Add support for migrate:undo