diff --git a/liblava/base/memory.cpp b/liblava/base/memory.cpp
index 0c00cd0dd9b9ace2e052f531c76e6a6c33cbb25f..fcc98e993219a28221e2eede580bd6b0c0a666bd 100644
--- a/liblava/base/memory.cpp
+++ b/liblava/base/memory.cpp
@@ -109,6 +109,9 @@ namespace lava {
 #if VMA_BIND_MEMORY2
             .vkBindBufferMemory2KHR = vkBindBufferMemory2KHR,
             .vkBindImageMemory2KHR = vkBindImageMemory2KHR,
+#endif
+#if VMA_MEMORY_BUDGET
+            .vkGetPhysicalDeviceMemoryProperties2KHR = vkGetPhysicalDeviceMemoryProperties2KHR
 #endif
         };