diff --git a/src/vkcv/ImageLayoutTransitions.cpp b/src/vkcv/ImageLayoutTransitions.cpp
index a64e06627868e0e0938ef791c1bd645ea1514114..0b08819489c41c5cde3ceddbb0629a5d2ae3cd30 100644
--- a/src/vkcv/ImageLayoutTransitions.cpp
+++ b/src/vkcv/ImageLayoutTransitions.cpp
@@ -1,7 +1,7 @@
 #include "ImageLayoutTransitions.hpp"
 
 namespace vkcv {
-	void transitionImageLayoutImmediate(const vk::CommandBuffer cmdBuffer, const VkImage image,
+	void transitionImageLayoutImmediate(const vk::CommandBuffer cmdBuffer, const vk::Image image,
 		const vk::ImageLayout oldLayout, const vk::ImageLayout newLayout) {
 
 		// TODO: proper src and dst masks
diff --git a/src/vkcv/ImageLayoutTransitions.hpp b/src/vkcv/ImageLayoutTransitions.hpp
index 284fb64c25a450195f4f1a54c7d4581a3fd80415..3dbfbdf6690a0683b30a96f400e7e4b6ec25c379 100644
--- a/src/vkcv/ImageLayoutTransitions.hpp
+++ b/src/vkcv/ImageLayoutTransitions.hpp
@@ -2,6 +2,6 @@
 #include <vulkan/vulkan.hpp>
 
 namespace vkcv {
-	void transitionImageLayoutImmediate(const vk::CommandBuffer cmdBuffer, const VkImage image,
+	void transitionImageLayoutImmediate(const vk::CommandBuffer cmdBuffer, const vk::Image image,
 		const vk::ImageLayout oldLayout, const vk::ImageLayout newLayout);
 }
\ No newline at end of file