stream: Replace subtree drain with a single node drain