Remove unwanted do-block in migrate and rollback procedures