clean up code

1 files changed, 3 insertions, 2 deletions

diff --git a/test_msgnet/main.go b/test_msgnet/main.go
index fd561f7..c2ae70d 100644
--- a/test_msgnet/main.go
+++ b/test_msgnet/main.go
@@ -114,9 +114,8 @@ func connHandler(_conn *C.struct_msgnetwork_conn_t, connected C.bool, _ unsafe.P
         }
     } else {
         fmt.Printf("[%s] Disconnected, retrying.\n", name)
-        addr := conn.GetAddr()
         err := salticidae.NewError()
-        net.Connect(addr, &err)
+        net.Connect(conn.GetAddr(), &err)
     }
 }
 
@@ -170,4 +169,6 @@ func main() {
     ev_term.Add(salticidae.SIGTERM)
 
     ec.Dispatch()
+    alice.net.Stop()
+    bob.net.Stop()
 }