diff --git a/networking/src/test/java/ch/dissem/bitmessage/networking/NetworkHandlerTest.java b/networking/src/test/java/ch/dissem/bitmessage/networking/NetworkHandlerTest.java
index 53ce36d..512de72 100644
--- a/networking/src/test/java/ch/dissem/bitmessage/networking/NetworkHandlerTest.java
+++ b/networking/src/test/java/ch/dissem/bitmessage/networking/NetworkHandlerTest.java
@@ -120,7 +120,7 @@ public class NetworkHandlerTest {
                 mock(NetworkHandler.MessageListener.class),
                 10);
         t.join();
-        assertEquals(3, nodeInventory.getInventory().size());
+        assertInventorySize(3, nodeInventory);
         assertInventorySize(3, peerInventory);
     }
 
@@ -137,7 +137,7 @@ public class NetworkHandlerTest {
                 mock(NetworkHandler.MessageListener.class),
                 10);
         t.join();
-        assertEquals(2, nodeInventory.getInventory().size());
+        assertInventorySize(2, nodeInventory);
         assertInventorySize(2, peerInventory);
     }
 
@@ -153,7 +153,7 @@ public class NetworkHandlerTest {
                 mock(NetworkHandler.MessageListener.class),
                 10);
         t.join();
-        assertEquals(1, nodeInventory.getInventory().size());
+        assertInventorySize(1, nodeInventory);
         assertInventorySize(1, peerInventory);
     }