diff --git a/src/client/main.cc b/src/client/main.cc
index 29a6657..db4317a 100644
--- a/src/client/main.cc
+++ b/src/client/main.cc
@@ -2,6 +2,7 @@
 #include <cstring>
 #include <iostream>
 #include <netinet/in.h>
+#include <ostream>
 #include <sys/socket.h>
 #include <unistd.h>
 
@@ -39,7 +40,8 @@ int main() {
         cout << "Type message to server: ";
         cin >> message.content;
 
-        if (strcmp(message.content, "exit") == 0) {
+        if (message.content == "exit") {
+            cout << "BYE!" << endl;
             break;
         }