Fix byte count logic