Add test for rollback loop