diff --git a/reference_designs/IRON-examples/matrix_multiplication_scalar/run.lit b/reference_designs/IRON-examples/matrix_multiplication_scalar/run.lit
index 49cd75e360..bb92a10b48 100644
--- a/reference_designs/IRON-examples/matrix_multiplication_scalar/run.lit
+++ b/reference_designs/IRON-examples/matrix_multiplication_scalar/run.lit
@@ -3,9 +3,9 @@
 //
 // REQUIRES: ryzen_ai
 //
-// RUN: %python %S/aie2.py > ./aie.mlir
+// RUN: %python %S/aie2.py ipu 0 > ./aie.mlir
 // RUN: %python aiecc.py --aie-generate-cdo --aie-generate-ipu --no-compile-host --xclbin-name=aie.xclbin --ipu-insts-name=insts.txt ./aie.mlir
 // RUN: clang %S/test.cpp -o test.exe -std=c++11 -Wall %xrt_flags -lrt -lstdc++ -lboost_program_options -lboost_filesystem
 // RUN: %run_on_ipu ./test.exe -x aie.xclbin -k MLIR_AIE -i insts.txt | FileCheck %s
 // CHECK: PASS!
-
+// XFAIL: *