Cleanup DFStage a little bit.