1 files changed, 3 insertions, 4 deletions

diff --git a/auto_tests/send_message_test.c b/auto_tests/send_message_test.c
index 8857d232..81969d59 100644
--- a/auto_tests/send_message_test.c
+++ b/auto_tests/send_message_test.c
@@ -11,6 +11,8 @@
 
 typedef struct State {
     uint32_t index;
+    uint64_t clock;
+
     bool message_received;
 } State;
 
@@ -51,10 +53,7 @@ static void send_message_test(Tox **toxes, State *state)
     ck_assert_msg(errm == TOX_ERR_FRIEND_SEND_MESSAGE_OK, "TOX_MAX_MESSAGE_LENGTH is too big? error=%d", errm);
 
     do {
-        tox_iterate(toxes[0], &state[0]);
-        tox_iterate(toxes[1], &state[1]);
-
-        c_sleep(ITERATION_INTERVAL);
+        iterate_all_wait(2, toxes, state, ITERATION_INTERVAL);
     } while (!state[1].message_received);
 }