diff --git a/tests/integration/cases/replace_validator.py b/tests/integration/cases/replace_validator.py
index 82009b5..97914be 100644
--- a/tests/integration/cases/replace_validator.py
+++ b/tests/integration/cases/replace_validator.py
@@ -54,12 +54,12 @@
                 "from pydantic import BaseModel, Field",
                 "",
                 "",
-                "class A(str, Enum):",
+                "class E(str, Enum):",
                 "    a = 'a'",
                 "    b = 'b'",
                 "",
                 "class A(BaseModel):",
-                "    a: A = Field(A.a, const=True)",
+                "    a: E = Field(E.a, const=True)",
             ],
         ),
         expected=File(
@@ -70,12 +70,12 @@
                 "from typing import Literal",
                 "",
                 "",
-                "class A(str, Enum):",
+                "class E(str, Enum):",
                 "    a = 'a'",
                 "    b = 'b'",
                 "",
                 "class A(BaseModel):",
-                "    a: Literal[A.a] = A.a",
+                "    a: Literal[E.a] = E.a",
             ],
         ),
     ),