fix custom stream demo