Fix wrap_cryptopro's sbox handling

[pygost.git] / pygost / wrap.py

diff --git a/pygost/wrap.py b/pygost/wrap.py
index 9deeff591f49647cc11afb19c5d8af548d5d54e0..eb3855fb8ce3d7f25c18f01ae53c61df610f1783 100644 (file)
--- a/pygost/wrap.py
+++ b/pygost/wrap.py
@@ -79,7 +79,12 @@ def wrap_cryptopro(ukm, kek, cek, sbox=DEFAULT_SBOX):
     :returns: wrapped key
     :rtype: bytes, 44 bytes
     """
-    return wrap_gost(ukm, diversify(kek, bytearray(ukm)), cek, sbox=sbox)
+    return wrap_gost(
+        ukm,
+        diversify(kek, bytearray(ukm), sbox=sbox),
+        cek,
+        sbox=sbox,
+    )
 
 
 def unwrap_cryptopro(kek, data, sbox=DEFAULT_SBOX):