Not actually sure what I changed here. But only DAGModule changed, so it
was likely quite minor.