add missing keeps after rematerializing nodes