add tests for rollback