Fix tests for old streaming impl.