diff --git a/src/commonMain/kotlin/net/sergeych/crypto2/kdf.kt b/src/commonMain/kotlin/net/sergeych/crypto2/kdf.kt
index e99dc12..76c3fc4 100644
--- a/src/commonMain/kotlin/net/sergeych/crypto2/kdf.kt
+++ b/src/commonMain/kotlin/net/sergeych/crypto2/kdf.kt
@@ -47,9 +47,9 @@ sealed class KDF {
         fun kdfForSize(numberOfKeys: Int,salt: UByteArray = Argon.randomSalt()): KDF =
             creteDefault(SymmetricKey.keyLength * numberOfKeys, this, salt)
 
-        fun kdfForSizeInBytes(sizeInBytes: Int, complexity: Complexity, domain: String): KDF {
+        fun kdfForSizeInBytes(sizeInBytes: Int, domain: String): KDF {
             val salt = Hash.Blake2b.deriveSalt(domain, Argon.saltSize)
-            return creteDefault(sizeInBytes, complexity, salt)
+            return creteDefault(sizeInBytes, this, salt)
         }
 
 
diff --git a/src/commonTest/kotlin/KDFTest.kt b/src/commonTest/kotlin/KDFTest.kt
index a19385c..4033ae7 100644
--- a/src/commonTest/kotlin/KDFTest.kt
+++ b/src/commonTest/kotlin/KDFTest.kt
@@ -66,7 +66,7 @@ class KDFTest {
         val domain = "kdf-size-in-bytes-test"
         val expectedSalt = Hash.Blake2b.deriveSalt(domain, KDF.Argon.saltSize)
 
-        val kdf = KDF.Complexity.Sensitive.kdfForSizeInBytes(size, KDF.Complexity.FixedLow, domain)
+        val kdf = KDF.Complexity.FixedLow.kdfForSizeInBytes(size, domain)
         assertIs<KDF.Argon>(kdf)
 
         assertEquals(KDF.Argon.create(KDF.Complexity.FixedLow, expectedSalt, size), kdf)