fix import_export_public_key test
diff --git a/tests/suites/test_suite_psa_crypto.function b/tests/suites/test_suite_psa_crypto.function
index ae208be..6458aa3 100644
--- a/tests/suites/test_suite_psa_crypto.function
+++ b/tests/suites/test_suite_psa_crypto.function
@@ -532,6 +532,7 @@
 void import_export_public_key( char *hex,
                                 int type_arg,
                                 int expected_bits,
+                                int public_key_expected_length,
                                 int expected_export_status )
 {
     int slot = 1;
@@ -572,7 +573,7 @@
         goto destroy;
 
  
-    TEST_ASSERT( exported_length == data_size );
+    TEST_ASSERT( exported_length == public_key_expected_length );
 
 destroy:
     /* Destroy the key */