diff --git a/mlir/test/python/dialects/affine.py b/mlir/test/python/dialects/affine.py
index 7ef128c1724c4..a944852a8b55f 100644
--- a/mlir/test/python/dialects/affine.py
+++ b/mlir/test/python/dialects/affine.py
@@ -333,3 +333,8 @@ def simple_affine_if_else(cond_operands):
             affine.AffineYieldOp([x_false, y_false])
         add = arith.AddIOp(if_op.results[0], if_op.results[1])
         return
+
+
+@constructAndPrintInModule
+def test_double_AtomicRMWKindAttr_registration():
+    from mlir.dialects import _affine_enum_gen