diff --git a/tests/suites/test_suite_pkwrite.function b/tests/suites/test_suite_pkwrite.function
index 11bc5a8..1697a51 100644
--- a/tests/suites/test_suite_pkwrite.function
+++ b/tests/suites/test_suite_pkwrite.function
@@ -38,7 +38,7 @@
         }
     } else {
         TEST_EQUAL(mbedtls_pk_parse_keyfile(&key, key_file, NULL,
-                                             mbedtls_test_rnd_std_rand, NULL), 0);
+                                            mbedtls_test_rnd_std_rand, NULL), 0);
         if (is_der) {
             ret = mbedtls_pk_write_key_der(&key, buf, check_buf_len);
         } else {
@@ -56,7 +56,7 @@
         start_buf = buf + check_buf_len - buf_len;
     } else {
         TEST_EQUAL(ret, 0);
-        buf_len = check_buf_len;
+        buf_len = strlen((char *) buf) + 1; /* +1 takes the string terminator into account */
         start_buf = buf;
     }