Adds rollback to 2.1 migration