diff --git a/src/villas-pipe.cpp b/src/villas-pipe.cpp
index 9d62b68cb..59a3cb2ba 100644
--- a/src/villas-pipe.cpp
+++ b/src/villas-pipe.cpp
@@ -133,9 +133,9 @@ static void * send_loop(void *ctx)
 	struct sample *smps[node->out.vectorize];
 
 	/* Initialize memory */
-	unsigned pool_cnt = (strcmp(node_type_name(node->_vt), "infiniband") == 0 ? 8192 : LOG2_CEIL(node->out.vectorize));
+	unsigned pool_size = node_type(node)->pool_size ? node_type(node)->pool_size : LOG2_CEIL(node->out.vectorize));
 
-	ret = pool_init(&sendd.pool, pool_cnt, SAMPLE_LENGTH(DEFAULT_SAMPLE_LENGTH), node_memory_type(node, &memory_hugepage));
+	ret = pool_init(&sendd.pool, pool_size, SAMPLE_LENGTH(DEFAULT_SAMPLE_LENGTH), node_memory_type(node, &memory_hugepage));
 
 	if (ret < 0)
 		error("Failed to allocate memory for receive pool.");
@@ -203,9 +203,9 @@ static void * recv_loop(void *ctx)
 	struct sample *smps[node->in.vectorize];
 
 	/* Initialize memory */
-	unsigned pool_cnt = (strcmp(node_type_name(node->_vt), "infiniband") == 0 ? 8192 : LOG2_CEIL(node->in.vectorize));
+	unsigned pool_size = node_type(node)->pool_size ? node_type(node)->pool_size : LOG2_CEIL(node->in.vectorize));
 
-	ret = pool_init(&recvv.pool, pool_cnt, SAMPLE_LENGTH(DEFAULT_SAMPLE_LENGTH), node_memory_type(node, &memory_hugepage));
+	ret = pool_init(&recvv.pool, pool_size, SAMPLE_LENGTH(DEFAULT_SAMPLE_LENGTH), node_memory_type(node, &memory_hugepage));
 
 	if (ret < 0)
 		error("Failed to allocate memory for receive pool.");