diff --git a/pygost/wrap.py b/pygost/wrap.py
index 9deeff5..cbb8bcf 100644
--- a/pygost/wrap.py
+++ b/pygost/wrap.py
@@ -79,7 +79,7 @@ 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):