diff --git a/backends/vulkan/test/test_vulkan_delegate.py b/backends/vulkan/test/test_vulkan_delegate.py
index a59d72a9ca0..9e4f9803685 100644
--- a/backends/vulkan/test/test_vulkan_delegate.py
+++ b/backends/vulkan/test/test_vulkan_delegate.py
@@ -123,7 +123,9 @@ def run_test(memory_layout):
                 program, compile_config=self._edge_compile_config
             )
 
-            edge_program = edge_program.transform([I64toI32(), MeanToSumDiv()])
+            edge_program = edge_program.transform(
+                [I64toI32(self._edge_compile_config._skip_dim_order), MeanToSumDiv()]
+            )
 
             edge_program = edge_program.to_backend(VulkanPartitioner(compile_options))