diff --git a/.github/workflows/cpp_examples.yml b/.github/workflows/cpp_examples.yml
index ecc21fd3b..fa3d5ae17 100644
--- a/.github/workflows/cpp_examples.yml
+++ b/.github/workflows/cpp_examples.yml
@@ -30,14 +30,4 @@ jobs:
         configure-options: -DKOMPUTE_OPT_USE_BUILT_IN_VULKAN_HEADER=ON KOMPUTE_OPT_FROM_SOURCE=ON
         build-options: --parallel # Given we don't build too many resources we can leverage parallel
     - name: Run tests
-      run: |
-        # Check that the expected output is printed
-        TEST_STDOUT=$(./examples/array_multiplication/build/src/kompute_array_mult)
-        echo "$TEST_STDOUT"
-        TEST_CODE=$?
-        echo "Exit value for test run: $TEST_CODE"
-        echo $TEST_STDOUT | grep -q "Output: {  0  4  12  }"
-        TEST_MATCH=$?
-        echo "Exit value for text match: $TEST_MATCH"
-        exit $(($TEST_CODE + $TEST_MATCH))
-
+      run: ./examples/array_multiplication/build/src/kompute_array_mult
diff --git a/examples/array_multiplication/src/main.cpp b/examples/array_multiplication/src/main.cpp
index 56e1c06d2..c8637577b 100644
--- a/examples/array_multiplication/src/main.cpp
+++ b/examples/array_multiplication/src/main.cpp
@@ -38,4 +38,8 @@ main()
         std::cout << elem << "  ";
     }
     std::cout << "}" << std::endl;
+
+    if (tensorOut->vector() != std::vector<float>{ 0, 4, 12 }) {
+        throw std::runtime_error("Result does not match");
+    }
 }