fix: Rollback wouldn't occured if there was no associated rollback function

fix: Rollback wouldn't occured if there was no associated rollback function #37