diff --git a/example/client.test.cpp b/example/client.test.cpp
index 26a3d53..beeaef0 100644
--- a/example/client.test.cpp
+++ b/example/client.test.cpp
@@ -45,8 +45,4 @@ int main(int argc, char* argv[]) {
     }
 
     return 0;
-}
-
-bool TCPClient::shouldStop() const {
-    return !running;
 }
\ No newline at end of file
diff --git a/src/TCPClient.cpp b/src/TCPClient.cpp
index f055a13..6f5b543 100644
--- a/src/TCPClient.cpp
+++ b/src/TCPClient.cpp
@@ -79,4 +79,8 @@ void TCPClient::start() {
 void TCPClient::stop() {
     running = false;
     close(clientSocket);
+}
+
+bool TCPClient::shouldStop() const {
+    return !running;
 }
\ No newline at end of file