diff --git a/tests/torture_key.c b/tests/torture_key.c
index c0ec845a..cdfb82c7 100644
--- a/tests/torture_key.c
+++ b/tests/torture_key.c
@@ -307,7 +307,7 @@ static const char torture_ed25519_public_testkey[] =
 
 static const char *torture_get_testkey_internal(enum ssh_keytypes_e type,
                                                 int bits,
-                                                int with_passphrase,
+                                                bool with_passphrase,
                                                 int pubkey)
 {
     switch (type) {
@@ -369,7 +369,7 @@ static const char *torture_get_testkey_internal(enum ssh_keytypes_e type,
 
 const char *torture_get_testkey(enum ssh_keytypes_e type,
                                 int ecda_bits,
-                                int with_passphrase)
+                                bool with_passphrase)
 {
     return torture_get_testkey_internal(type, ecda_bits, with_passphrase, 0);
 }
diff --git a/tests/torture_key.h b/tests/torture_key.h
index f5f54250..7a476e73 100644
--- a/tests/torture_key.h
+++ b/tests/torture_key.h
@@ -24,11 +24,13 @@
 #ifndef _TORTURE_KEY_H
 #define _TORTURE_KEY_H
 
+#include <stdbool.h>
+
 #define TORTURE_TESTKEY_PASSWORD "libssh-rocks"
 
 const char *torture_get_testkey(enum ssh_keytypes_e type,
                                 int ecdsa_bits,
-                                int with_passphrase);
+                                bool with_passphrase);
 const char *torture_get_testkey_passphrase(void);
 
 const char *torture_get_testkey_pub(enum ssh_keytypes_e type, int ecdsa_bits);