diff --git a/dev/ast-grep/rules/remove-nullable-arg.yaml b/dev/ast-grep/rules/remove-nullable-arg.yaml
index b101d4b2e4..5ee2f32dcc 100644
--- a/dev/ast-grep/rules/remove-nullable-arg.yaml
+++ b/dev/ast-grep/rules/remove-nullable-arg.yaml
@@ -3,8 +3,8 @@ language: python
 rule:
   pattern: $X = mapped_column($$$ARGS)
   any:
-    - pattern: $X = mapped_column($$$BEFORE, String, $$$MID, nullable=True, $$$AFTER)
-    - pattern: $X = mapped_column($$$BEFORE, String, $$$MID, nullable=True)
+    - pattern: $X = mapped_column($$$BEFORE, sa.String, $$$MID, nullable=True, $$$AFTER)
+    - pattern: $X = mapped_column($$$BEFORE, sa.String, $$$MID, nullable=True)
 rewriters:
 - id: filter-string-nullable
   rule: