diff --git a/renderdoc/driver/vulkan/wrappers/vk_resource_funcs.cpp b/renderdoc/driver/vulkan/wrappers/vk_resource_funcs.cpp
index e6a6111ab..853c46bc3 100644
--- a/renderdoc/driver/vulkan/wrappers/vk_resource_funcs.cpp
+++ b/renderdoc/driver/vulkan/wrappers/vk_resource_funcs.cpp
@@ -184,7 +184,8 @@ VkResult WrappedVulkan::vkAllocMemory(
 			VkDeviceMemory*                             pMem)
 {
 	VkMemoryAllocInfo info = *pAllocInfo;
-	info.memoryTypeIndex = GetRecord(device)->memIdxMap[info.memoryTypeIndex];
+	if(m_State >= WRITING)
+		info.memoryTypeIndex = GetRecord(device)->memIdxMap[info.memoryTypeIndex];
 	VkResult ret = ObjDisp(device)->AllocMemory(Unwrap(device), &info, pMem);
 	
 	if(ret == VK_SUCCESS)