diff --git a/test/python/dialects/affine.py b/test/python/dialects/affine.py
index 832ec37..344336a 100644
--- a/test/python/dialects/affine.py
+++ b/test/python/dialects/affine.py
@@ -47,12 +47,10 @@ def affine_store_test(arg0):
 # CHECK-LABEL: TEST: testAffineDelinearizeInfer
 @constructAndPrintInModule
 def testAffineDelinearizeInfer():
-    # CHECK: %[[C0:.*]] = arith.constant 0 : index
-    c0 = arith.ConstantOp(T.index(), 0)
     # CHECK: %[[C1:.*]] = arith.constant 1 : index
     c1 = arith.ConstantOp(T.index(), 1)
-    # CHECK: %{{.*}}:2 = affine.delinearize_index %[[C1:.*]] into (%[[C1:.*]], %[[C0:.*]]) : index, index
-    two_indices = affine.AffineDelinearizeIndexOp(c1, [c1, c0])
+    # CHECK: %{{.*}}:2 = affine.delinearize_index %[[C1:.*]] into (2, 3) : index, index
+    two_indices = affine.AffineDelinearizeIndexOp(c1, [], [2, 3])
 
 
 # CHECK-LABEL: TEST: testAffineLoadOp
@@ -159,7 +157,7 @@ def testAffineForOpErrors():
         )
 
     try:
-        two_indices = affine.AffineDelinearizeIndexOp(c1, [c1, c1])
+        two_indices = affine.AffineDelinearizeIndexOp(c1, [], [1, 1])
         affine.AffineForOp(
             two_indices,
             c2,
@@ -265,3 +263,4 @@ def range_loop_8(lb, ub, memref_v):
             add = arith.addi(i, i)
             memref.store(add, it, [i])
             affine.yield_([it])
+