Fix IO bug