diff --git a/mlir/test/python/execution_engine.py b/mlir/test/python/execution_engine.py
--- a/mlir/test/python/execution_engine.py
+++ b/mlir/test/python/execution_engine.py
@@ -370,8 +370,9 @@
         %now = call @nano_time() : () -> i64
         %memref = memref.alloca() : memref<1xi64>
         %c0 = arith.constant 0 : index
-        memref.store %memref[%c0] : memref<1xi64>
-        call @print_memref_i64(%memref) : (memref<*xi64) -> ()
+        memref.store %now, %memref[%c0] : memref<1xi64>
+        %u_memref = memref.cast %memref : memref<1xi64> to memref<*xi64>
+        call @print_memref_i64(%u_memref) : (memref<*xi64>) -> ()
         return
       }
       func private @nano_time() -> i64 attributes { llvm.emit_c_interface }
@@ -385,7 +386,7 @@
             "../../../../lib/libmlir_runner_utils.so",
             "../../../../lib/libmlir_c_runner_utils.so"
         ])
-    execution_engine.invoke("main", arg0_memref_ptr)
+    execution_engine.invoke("main")
     # CHECK: Unranked Memref
     # CHECK: [{{.*}}]