diff --git a/unit_tests/util/query_heap.cpp b/unit_tests/util/query_heap.cpp
index bca90fc73..a56b3c78e 100644
--- a/unit_tests/util/query_heap.cpp
+++ b/unit_tests/util/query_heap.cpp
@@ -23,7 +23,7 @@ using TestNodeID = NodeID;
 using TestKey = int;
 using TestWeight = int;
 using storage_types = boost::mpl::list<ArrayStorage<TestNodeID, TestKey>,
-                                       MapStorage<TestNodeID, TestKey>,
+                                       TwoLevelStorage<TestNodeID, TestKey>,
                                        UnorderedMapStorage<TestNodeID, TestKey>>;
 
 template <unsigned NUM_ELEM> struct RandomDataFixture