Fix assert arguments in LMS and LMOTS export tests

Signed-off-by: Raef Coles <raef.coles@arm.com>
diff --git a/tests/suites/test_suite_lmots.function b/tests/suites/test_suite_lmots.function
index 8857326..f18b79c 100644
--- a/tests/suites/test_suite_lmots.function
+++ b/tests/suites/test_suite_lmots.function
@@ -163,9 +163,10 @@
                                                    exported_pub_key_buf_size,
                                                    &exported_pub_key_size ), 0 );
 
-        TEST_EQUAL( exported_pub_key_buf_size, exported_pub_key_size );
-        ASSERT_COMPARE( pub_key->x, MBEDTLS_LMOTS_PUBLIC_KEY_LEN(MBEDTLS_LMOTS_SHA256_N32_W8),
-                        exported_pub_key, MBEDTLS_LMOTS_PUBLIC_KEY_LEN(MBEDTLS_LMOTS_SHA256_N32_W8) );
+        TEST_EQUAL( exported_pub_key_buf_size,
+                    MBEDTLS_LMOTS_PUBLIC_KEY_LEN(MBEDTLS_LMOTS_SHA256_N32_W8) );
+        ASSERT_COMPARE( pub_key->x, pub_key->len,
+                        exported_pub_key, exported_pub_key_size );
         mbedtls_free(exported_pub_key);
         exported_pub_key = NULL;
 
diff --git a/tests/suites/test_suite_lms.function b/tests/suites/test_suite_lms.function
index b6ae309..97f45e6 100644
--- a/tests/suites/test_suite_lms.function
+++ b/tests/suites/test_suite_lms.function
@@ -165,9 +165,10 @@
                                                    exported_pub_key_buf_size,
                                                    &exported_pub_key_size ), 0 );
 
-        TEST_EQUAL( exported_pub_key_buf_size, exported_pub_key_size );
-        ASSERT_COMPARE( pub_key->x, MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10),
-                        exported_pub_key, MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10) );
+        TEST_EQUAL( exported_pub_key_size,
+                    MBEDTLS_LMS_PUBLIC_KEY_LEN(MBEDTLS_LMS_SHA256_M32_H10 ) );
+        ASSERT_COMPARE( pub_key->x, pub_key->len,
+                        exported_pub_key, exported_pub_key_size );
         mbedtls_free(exported_pub_key);
         exported_pub_key = NULL;