Replace take operation to drainTo