1 files changed, 5 insertions, 8 deletions
diff --git a/directory/src/main/java/com/orbekk/same/directory/DirectoryApp.java b/directory/src/main/java/com/orbekk/same/directory/DirectoryApp.java
index 7a05894..667d4f0 100644
--- a/directory/src/main/java/com/orbekk/same/directory/DirectoryApp.java
+++ b/directory/src/main/java/com/orbekk/same/directory/DirectoryApp.java
@@ -3,6 +3,7 @@ package com.orbekk.same.directory;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import com.orbekk.protobuf.SimpleProtobufServer;
 import com.orbekk.same.discovery.DirectoryService;
 import com.orbekk.same.http.JettyServerBuilder;
 import com.orbekk.same.http.ServerContainer;
@@ -12,13 +13,9 @@ public class DirectoryApp {
     
     public static void main(String[] args) {
         Logger logger = LoggerFactory.getLogger(DirectoryApp.class);
-        ServerContainer server = new JettyServerBuilder(DISCOVERY_PORT)
-            .withService(new DirectoryServiceImpl(), DirectoryService.class)
-            .build();
-        try {
-            server.start();
-        } catch (Exception e) {
-            logger.error("Unable to start server.", e);
-        }
+        SimpleProtobufServer server =
+                SimpleProtobufServer.create(DISCOVERY_PORT);
+        server.registerService(new DirectoryServiceImpl());
+        server.start();
     }
 }