Update rollback behavior