Update for rollback infrastructure