diff --git a/main.cc b/main.cc
index 898842e..dc8bcd4 100644
--- a/main.cc
+++ b/main.cc
@@ -126,7 +126,12 @@ main(int argc, char *argv[])
 	}
 
 	endwin();
-	closeServer();
+	//closeServer();
+
+	if (mode == 1)
+		closeServer();
+	else if (mode == 2)
+		closeClient();
 
 	return 0;
 }