diff --git a/test/python/aie_ops.py b/test/python/aie_ops.py
index be9a056c2d..83d2488954 100644
--- a/test/python/aie_ops.py
+++ b/test/python/aie_ops.py
@@ -1,7 +1,7 @@
 # Copyright (C) 2022, Advanced Micro Devices, Inc.
 # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
-# RUN: python %s | FileCheck %s
+# RUN: python3 %s | FileCheck %s
 
 import aie
 from aie.mlir.ir import *