layers: Update GetImageState helper function

Change-Id: Ie79232fabe8410c1b709e3d65c25db90dbbf7884
diff --git a/layers/descriptor_sets.cpp b/layers/descriptor_sets.cpp
index 690fb66..046dac9 100644
--- a/layers/descriptor_sets.cpp
+++ b/layers/descriptor_sets.cpp
@@ -831,7 +831,7 @@
                         return false;
                     }
 
-                    auto image_node = device_data_->GetImageState(device_data_, image_view_ci.image);
+                    auto image_node = device_data_->GetImageState(image_view_ci.image);
                     assert(image_node);
                     // Verify Image Layout
                     // Copy first mip level into sub_layers and loop over each mip level to verify layout
@@ -1339,7 +1339,7 @@
     VkImage image = iv_state->create_info.image;
     VkFormat format = VK_FORMAT_MAX_ENUM;
     VkImageUsageFlags usage = 0;
-    auto image_node = dev_data->GetImageState(dev_data, image);
+    auto image_node = dev_data->GetImageState(image);
     if (image_node) {
         format = image_node->createInfo.format;
         usage = image_node->createInfo.usage;