diff --git a/example/client.test.cpp b/example/client.test.cpp
index 10e9e21..26a3d53 100644
--- a/example/client.test.cpp
+++ b/example/client.test.cpp
@@ -46,3 +46,7 @@ int main(int argc, char* argv[]) {
 
     return 0;
 }
+
+bool TCPClient::shouldStop() const {
+    return !running;
+}
\ No newline at end of file
diff --git a/include/TCPSocket/TCPClient.hpp b/include/TCPSocket/TCPClient.hpp
index b8e1b1e..b002bfd 100644
--- a/include/TCPSocket/TCPClient.hpp
+++ b/include/TCPSocket/TCPClient.hpp
@@ -32,4 +32,5 @@ public:
 
     virtual void handleMessage(const std::string& message);
 
+    bool shouldStop() const;
 };
\ No newline at end of file