Fix broken pipe handling