1 files changed, 5 insertions, 6 deletions
diff --git a/src/cmd/tls_server.cpp b/src/cmd/tls_server.cpp
index 9db250859..fd9f7ef5d 100644
--- a/src/cmd/tls_server.cpp
+++ b/src/cmd/tls_server.cpp
@@ -15,9 +15,6 @@ using namespace Botan;
 
 using namespace std::placeholders;
 
-#include <string>
-#include <iostream>
-#include <memory>
 #include <list>
 
 #include <sys/types.h>
@@ -120,9 +117,7 @@ void alert_received(TLS::Alert alert, const byte[], size_t)
    std::cout << "Alert: " << alert.type_string() << "\n";
    }
 
-}
-
-int tls_server_main(int argc, char* argv[])
+int tls_server(int argc, char* argv[])
    {
    int port = 4433;
    std::string transport = "tcp";
@@ -262,4 +257,8 @@ int tls_server_main(int argc, char* argv[])
    return 0;
    }
 
+REGISTER_APP(tls_server);
+
+}
+
 #endif