1 files changed, 8 insertions, 5 deletions

diff --git a/testing/Lossless_UDP_testclient.c b/testing/Lossless_UDP_testclient.c
index 241f5bdc..bd4b1e23 100644
--- a/testing/Lossless_UDP_testclient.c
+++ b/testing/Lossless_UDP_testclient.c
@@ -68,12 +68,15 @@ int main(int argc, char *argv[])
     uint64_t timer = current_time();
     while(1)
     {
-        if(is_connected(connection) != 2)
+
+        if(is_connected(connection) == 3)
         {
-            if(is_connected(connection) == 1)
-            {
-                printf("Connecting took: %llu us", (unsigned long long)(current_time() - timer));
-            }
+            printf("Connecting took: %llu us", (unsigned long long)(current_time() - timer));
+            break;
+        }
+        if(is_connected(connection) == 0)
+        {
+            printf("Connection timeout after: %llu us", (unsigned long long)(current_time() - timer));
             break;
         }
         c_sleep(1);