diff --git a/src/util/top_sort.h b/src/util/top_sort.h
index 2943b68b6..0f2c32cb7 100644
--- a/src/util/top_sort.h
+++ b/src/util/top_sort.h
@@ -44,7 +44,7 @@ class top_sort {
 
 
     bool contains_partition(T* f) const {
-        return m_partition_id.get(f->get_small_id()) != UINT_MAX;
+        return m_partition_id.get(f->get_small_id(), UINT_MAX) != UINT_MAX;
     }