diff --git a/computecloth/computecloth.cpp b/computecloth/computecloth.cpp
index 9305fe89..0f958897 100644
--- a/computecloth/computecloth.cpp
+++ b/computecloth/computecloth.cpp
@@ -242,7 +242,7 @@ public:
 			bufferBarriers.push_back(bufferBarrier);
 
 			vkCmdPipelineBarrier(compute.commandBuffers[i],
-				VK_PIPELINE_STAGE_VERTEX_SHADER_BIT,
+				VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
 				VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
 				VK_FLAGS_NONE,
 				0, nullptr,
@@ -295,7 +295,7 @@ public:
 			vkCmdPipelineBarrier(
 				compute.commandBuffers[i],
 				VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
-				VK_PIPELINE_STAGE_VERTEX_SHADER_BIT,
+				VK_PIPELINE_STAGE_COMPUTE_SHADER_BIT,
 				VK_FLAGS_NONE,
 				0, nullptr,
 				static_cast<uint32_t>(bufferBarriers.size()), bufferBarriers.data(),
diff --git a/data/shaders/computecloth/cloth.comp.spv b/data/shaders/computecloth/cloth.comp.spv
index 98bba249..770aa432 100644
Binary files a/data/shaders/computecloth/cloth.comp.spv and b/data/shaders/computecloth/cloth.comp.spv differ