Minor edit to rollback script.