diff --git a/src/commonMain/kotlin/com/infendro/random/HMAC.kt b/src/commonMain/kotlin/com/infendro/random/HMAC.kt
index 0707679..d22fbae 100644
--- a/src/commonMain/kotlin/com/infendro/random/HMAC.kt
+++ b/src/commonMain/kotlin/com/infendro/random/HMAC.kt
@@ -37,11 +37,13 @@ class HMAC internal constructor(
                 V = hmac.hash(K, V)
                 addAll(V)
             }
-        }.take(count).toByteArray()
+        }
 
         K = hmac.hash(K, V + byteArrayOf(0x00))
         V = hmac.hash(K, V)
 
         return bytes
+            .take(count)
+            .toByteArray()
     }
 }