diff --git a/lang/java/reef-io/src/test/java/org/apache/reef/io/network/NetworkServiceTest.java b/lang/java/reef-io/src/test/java/org/apache/reef/io/network/NetworkServiceTest.java
index 373fd31..86acc86 100644
--- a/lang/java/reef-io/src/test/java/org/apache/reef/io/network/NetworkServiceTest.java
+++ b/lang/java/reef-io/src/test/java/org/apache/reef/io/network/NetworkServiceTest.java
@@ -386,7 +386,7 @@
 
           final Injector injectorNs2 = injector2.forkInjector();
           injectorNs2.bindVolatileParameter(NetworkServiceParameters.NetworkServiceHandler.class,
-              new MessageHandler<String>(name2, monitor, numMessages));
+              new MessageHandler<String>(name2, monitor, totalNumMessages));
           final NetworkService<String> ns2 = injectorNs2.getInstance(NetworkService.class);
 
           final Injector injectorNs1 = injector2.forkInjector();