Fix rollback test