From 479ef98ec64451ebe92da9e0899175213492af65 Mon Sep 17 00:00:00 2001
From: Jae Kwon <jae@tendermint.com>
Date: Sun, 21 Feb 2016 23:44:43 -0800
Subject: [PATCH] Conform to TMSP NewServer

---
 cmd/main.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cmd/main.go b/cmd/main.go
index 1afc7235e..c0b406f41 100644
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -48,7 +48,7 @@ func main() {
 	}
 
 	// Start the listener
-	_, err = server.StartListener(*addrPtr, app)
+	svr, err := server.NewServer(*addrPtr, app)
 	if err != nil {
 		Exit("create listener: " + err.Error())
 	}
@@ -56,6 +56,7 @@ func main() {
 	// Wait forever
 	TrapSignal(func() {
 		// Cleanup
+		svr.Stop()
 	})
 
 }