Fix streaming test failures