diff --git a/.github/workflows/trunk.yml b/.github/workflows/trunk.yml
index 5a6720cdfad..cca1fe5fe45 100644
--- a/.github/workflows/trunk.yml
+++ b/.github/workflows/trunk.yml
@@ -278,6 +278,7 @@ jobs:
       matrix:
         include:
           - test_arm_backend: test_pytest_ops_vkml
+          - test_arm_backend: test_pytest_models_vkml
           - test_arm_backend: test_ootb_tests_vgf
       fail-fast: false
     with:
diff --git a/backends/arm/test/models/test_swin2sr_arm.py b/backends/arm/test/models/test_swin2sr_arm.py
index e4fc6f07950..5fd29943b94 100644
--- a/backends/arm/test/models/test_swin2sr_arm.py
+++ b/backends/arm/test/models/test_swin2sr_arm.py
@@ -42,6 +42,9 @@
     "executorch_exir_dialects_edge__ops_quantized_decomposed_dequantize_per_tensor_default": 5,
     "executorch_exir_dialects_edge__ops_quantized_decomposed_quantize_per_tensor_default": 6,
 }
+swin2sr_vgf_quant_lowered_outer_graph_ops = {
+    "torch.ops.higher_order.executorch_call_delegate": 1,
+}
 
 
 class TinySwin2SR(torch.nn.Module):
@@ -110,7 +113,7 @@ def test_swin2sr_vgf_quant():
         quantize=True,
         run_on_vulkan_runtime=sys.platform == "linux",
     )
-    pipeline.change_args("check_count.exir", swin2sr_int_lowered_outer_graph_ops)
+    pipeline.change_args("check_count.exir", swin2sr_vgf_quant_lowered_outer_graph_ops)
     pipeline.run()