diff --git a/tests/testsuite.c b/tests/testsuite.c
index 14335a75..f8d4c674 100644
--- a/tests/testsuite.c
+++ b/tests/testsuite.c
@@ -47,20 +47,18 @@
     #include "tests/sftp.h"
 #endif
 
-#ifndef NO_TESTSUITE_MAIN_DRIVER
-
-static int TestsuiteTest(int argc, char** argv);
+#if !defined(NO_TESTSUITE_MAIN_DRIVER) && !defined(NO_MAIN_FUNCTION)
 
 int main(int argc, char** argv)
 {
-    return TestsuiteTest(argc, argv);
+    return wolfSSH_TestsuiteTest(argc, argv);
 }
 
 
 int myoptind = 0;
 char* myoptarg = NULL;
 
-#endif /* NO_TESTSUITE_MAIN_DRIVER */
+#endif /* !NO_TESTSUITE_MAIN_DRIVER && !NO_MAIN_FUNCTION */
 
 
 #if !defined(NO_WOLFSSH_SERVER) && !defined(NO_WOLFSSH_CLIENT)
@@ -87,7 +85,7 @@ static int tsClientUserAuth(byte authType, WS_UserAuthData* authData, void* ctx)
 #define NUMARGS 5
 #define ARGLEN 32
 
-int TestsuiteTest(int argc, char** argv)
+int wolfSSH_TestsuiteTest(int argc, char** argv)
 {
     tcp_ready ready;
     THREAD_TYPE serverThread;
diff --git a/tests/testsuite.h b/tests/testsuite.h
index 1b9d3761..9710c77d 100644
--- a/tests/testsuite.h
+++ b/tests/testsuite.h
@@ -18,7 +18,9 @@
  * along with wolfSSH.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-#pragma once
+#ifndef _WOLFSSH_TESTS_TESTSUITE_H_
+#define _WOLFSSH_TESTS_TESTSUITE_H_
 
-int test_SFTP(int flag);
+int wolfSSH_TestsuiteTest(int argc, char** argv);
 
+#endif /* _WOLFSSH_TESTS_TESTSUITE_H_ */