Fix rollback behaviour.