Add rollback support to component