diff --git a/__tests__/benchmark.test.js b/__tests__/benchmark.test.js
index c2a28344ff5fb93cf258f9906892e994243b9354..a8c8ba58f7453577228823cd9fc7757d58f7dad7 100644
--- a/__tests__/benchmark.test.js
+++ b/__tests__/benchmark.test.js
@@ -198,7 +198,7 @@ describe("drawing app mesh", () => {
               broadcasts += 1
 
               // TODO: can we assume that we only use one message here?
-              if (broadcasts < ITERATIONS) {
+              if (broadcasts <= ITERATIONS) {
                 addRemTime +=
                   (currTime[0] - prevTime[0]) * 1e9 +
                   (currTime[1] - prevTime[1])
@@ -210,12 +210,12 @@ describe("drawing app mesh", () => {
 
               let packet
 
-              if (broadcasts <= ITERATIONS) {
+              if (broadcasts < ITERATIONS) {
                 packet = addPackets[broadcasts]
               } else if (broadcasts < ITERATIONS * 2) {
                 packet = erasePackets[broadcasts - ITERATIONS]
               } else {
-                resolve()
+                return resolve()
               }
 
               prevTime = process.hrtime()
@@ -877,7 +877,7 @@ describe("drawing app mesh", () => {
             const timeoutCallback = () => {
               broadcasts += 1
 
-              if (broadcasts < ITERATIONS) {
+              if (broadcasts <= ITERATIONS) {
                 addRemTime +=
                   (currTime[0] - prevTime[0]) * 1e9 +
                   (currTime[1] - prevTime[1])