diff --git a/tests/unit/test_replace_config.py b/tests/unit/test_replace_config.py
index 4fa4fc2..8c024a1 100644
--- a/tests/unit/test_replace_config.py
+++ b/tests/unit/test_replace_config.py
@@ -130,7 +130,7 @@ class Config(SuperConfig):
         """
         self.assertCodemod(before, after)
 
-    @pytest.mark.xfail(reason="Not implemented yet")
+    @pytest.mark.xfail(reason="Comments inside Config are swallowed.")
     def test_inner_comments(self) -> None:
         before = """
         from pydantic import BaseModel
@@ -176,7 +176,6 @@ class Config(SuperConfig):
         """
         self.assertCodemod(before, after)
 
-    @pytest.mark.xfail(reason="Not implemented yet")
     def test_extra_enum(self) -> None:
         before = """
         from pydantic import BaseModel, Extra
@@ -186,7 +185,7 @@ class Config:
                 extra = Extra.allow
         """
         after = """
-        from pydantic import BaseModel
+        from pydantic import ConfigDict, BaseModel
 
         class Potato(BaseModel):
             model_config = ConfigDict(extra="allow")