Fixed migration rollback bug