Layers: Fix screenshot seg fault, changed accesses to dispatch tbl
diff --git a/layers/screenshot.cpp b/layers/screenshot.cpp
index f7927df..19e8ed3 100644
--- a/layers/screenshot.cpp
+++ b/layers/screenshot.cpp
@@ -163,8 +163,8 @@
VkMemoryRequirements memRequirements;
uint32_t num_allocations = 0;
size_t num_alloc_size = sizeof(num_allocations);
- VkLayerDispatchTable* pTableDevice = screenshot_device_table_map[device];
- VkLayerDispatchTable* pTableQueue = screenshot_device_table_map[queue];
+ VkLayerDispatchTable* pTableDevice = get_dispatch_table(screenshot_device_table_map, device);
+ VkLayerDispatchTable* pTableQueue = get_dispatch_table(screenshot_device_table_map, queue);
VkLayerDispatchTable* pTableCmdBuffer;
if (imageMap.empty() || imageMap.find(image1) == imageMap.end())