diff --git a/plugin/trino-memory/src/test/java/io/trino/plugin/memory/MemoryQueryRunner.java b/plugin/trino-memory/src/test/java/io/trino/plugin/memory/MemoryQueryRunner.java
index 69c1b35274bc..ac4c0f750e7a 100644
--- a/plugin/trino-memory/src/test/java/io/trino/plugin/memory/MemoryQueryRunner.java
+++ b/plugin/trino-memory/src/test/java/io/trino/plugin/memory/MemoryQueryRunner.java
@@ -56,6 +56,7 @@ public static class Builder
             extends DistributedQueryRunner.Builder<Builder>
     {
         private List<TpchTable<?>> initialTables = ImmutableList.of();
+        private ImmutableMap.Builder<String, String> memoryProperties = ImmutableMap.builder();
 
         protected Builder()
         {
@@ -68,6 +69,19 @@ public Builder setInitialTables(Iterable<TpchTable<?>> initialTables)
             return self();
         }
 
+        public Builder setMemoryProperties(Map<String, String> memoryProperties)
+        {
+            this.memoryProperties = ImmutableMap.<String, String>builder()
+                    .putAll(requireNonNull(memoryProperties, "memoryProperties is null"));
+            return self();
+        }
+
+        public Builder addMemoryProperty(String key, String value)
+        {
+            this.memoryProperties.put(key, value);
+            return self();
+        }
+
         @Override
         public DistributedQueryRunner build()
                 throws Exception
@@ -76,7 +90,7 @@ public DistributedQueryRunner build()
 
             try {
                 queryRunner.installPlugin(new MemoryPlugin());
-                queryRunner.createCatalog(CATALOG, "memory", ImmutableMap.of());
+                queryRunner.createCatalog(CATALOG, "memory", memoryProperties.buildOrThrow());
 
                 queryRunner.installPlugin(new TpchPlugin());
                 queryRunner.createCatalog("tpch", "tpch", ImmutableMap.of());