diff --git a/libwvdrmengine/oemcrypto/test/oemcrypto_test.cpp b/libwvdrmengine/oemcrypto/test/oemcrypto_test.cpp
index 2c9e3f56..5b0086f2 100644
--- a/libwvdrmengine/oemcrypto/test/oemcrypto_test.cpp
+++ b/libwvdrmengine/oemcrypto/test/oemcrypto_test.cpp
@@ -2291,7 +2291,7 @@ TEST_F(OEMCryptoSessionTests, QueryKeyControl) {
   size_t size = sizeof(block) - 1;
   OEMCryptoResult sts = OEMCrypto_QueryKeyControl(
       s.session_id(), s.license().keys[0].key_id,
-      sizeof(s.license().keys[0].key_id), reinterpret_cast<uint8_t*>(&block),
+      s.license().keys[0].key_id_length, reinterpret_cast<uint8_t*>(&block),
       &size);
   if (sts == OEMCrypto_ERROR_NOT_IMPLEMENTED) {
     return;