diff --git a/renderdoc/driver/vulkan/wrappers/vk_queue_funcs.cpp b/renderdoc/driver/vulkan/wrappers/vk_queue_funcs.cpp
index e9a2d2628..ff17ba4d0 100644
--- a/renderdoc/driver/vulkan/wrappers/vk_queue_funcs.cpp
+++ b/renderdoc/driver/vulkan/wrappers/vk_queue_funcs.cpp
@@ -1102,6 +1102,10 @@ void WrappedVulkan::CaptureQueueSubmit(VkQueue queue,
     {
       (*it)->AddResourceReferences(GetResourceManager());
       GetResourceManager()->MergeReferencedMemory((*it)->cmdInfo->memFrameRefs);
+    }
+
+    for(auto it = cmdsWithReferences.begin(); it != cmdsWithReferences.end(); ++it)
+    {
       UpdateImageStates((*it)->cmdInfo->imageStates);
     }