fix ReadBitsToSlice logic v2v1.2.6

1 files changed, 10 insertions, 7 deletions

diff --git a/bitreader.go b/bitreader.go
index 267ff5e..e455d5d 100644
--- a/bitreader.go
+++ b/bitreader.go
@@ -327,18 +327,21 @@ func (reader *ReaderType) ReadBitsToSlice(bits int) ([]byte, error) {
         out := make([]byte, bytes)
         for i := 0; i < bytes; i++ {
                 if i == bytes-1 { // Not enough to fill a whole byte
-                        val, err := reader.ReadBits(bits % 8)
+                        if bits%8 != 0 {
+                                val, err := reader.ReadBits(bits % 8)
+                                if err != nil {
+                                        return out, err
+                                }
+                                out[i] = byte(val)
+                        }
+                        break
+                } else {
+                        val, err := reader.ReadBytes(1)
                         if err != nil {
                                 return out, err
                         }
                         out[i] = byte(val)
-                        break
                 }
-                val, err := reader.ReadBytes(1)
-                if err != nil {
-                        return out, err
-                }
-                out[i] = byte(val)
         }
         return out, nil
 }