Update dissect/util/stream.py

Co-authored-by: Stefan de Reuver <9864602+Horofic@users.noreply.github.com>
fox-it · Jan 15, 2024 · b947295 · b947295
1 parent eb7ea58
commit b947295
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dissect/util/stream.py b/dissect/util/stream.py
@@ -601,10 +601,10 @@ def _read_fh(self, length: int) -> bytes:
             offset = self._zlib_prepend_offset
             self._zlib_prepend_offset += length
             return self._zlib_prepend[offset : self._zlib_prepend_offset]
-        else:
-            offset = self._zlib_prepend_offset
-            self._zlib_prepend_offset = None
-            return self._zlib_prepend[offset:] + self._fh.read(length - len(self._zlib_prepend) + offset)
+
+        offset = self._zlib_prepend_offset
+        self._zlib_prepend_offset = None
+        return self._zlib_prepend[offset:] + self._fh.read(length - len(self._zlib_prepend) + offset)
 
     def _read_zlib(self, length: int) -> bytes:
         if length < 0: