Minor fix with receiving amount of byes