diff --git a/builtins.go b/builtins.go
index 60fa894..f0be6fc 100644
--- a/builtins.go
+++ b/builtins.go
@@ -19,10 +19,10 @@ var builtins = []Func{
 	{"(*encoding/xml.Decoder).DecodeElement", "xml", 0, []string{"encoding/xml.Unmarshaler", "encoding.TextUnmarshaler"}},
 
 	// https://pkg.go.dev/gopkg.in/yaml.v3
-	{"gopkg.in/yaml.v3.Marshal", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler"}},
-	{"gopkg.in/yaml.v3.Unmarshal", "yaml", 1, []string{"gopkg.in/yaml.v3.Unmarshaler"}},
-	{"(*gopkg.in/yaml.v3.Encoder).Encode", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler"}},
-	{"(*gopkg.in/yaml.v3.Decoder).Decode", "yaml", 0, []string{"gopkg.in/yaml.v3.Unmarshaler"}},
+	{"gopkg.in/yaml.v3.Marshal", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler", "encoding.TextMarshaler"}},
+	{"gopkg.in/yaml.v3.Unmarshal", "yaml", 1, []string{"gopkg.in/yaml.v3.Unmarshaler", "encoding.TextUnmarshaler"}},
+	{"(*gopkg.in/yaml.v3.Encoder).Encode", "yaml", 0, []string{"gopkg.in/yaml.v3.Marshaler", "encoding.TextMarshaler"}},
+	{"(*gopkg.in/yaml.v3.Decoder).Decode", "yaml", 0, []string{"gopkg.in/yaml.v3.Unmarshaler", "encoding.TextUnmarshaler"}},
 
 	// https://pkg.go.dev/github.com/BurntSushi/toml
 	{"github.com/BurntSushi/toml.Unmarshal", "toml", 1, []string{"github.com/BurntSushi/toml.Unmarshaler", "encoding.TextUnmarshaler"}},
diff --git a/testdata/src/tests/builtins.go b/testdata/src/tests/builtins.go
index 482839b..5fba83a 100644
--- a/testdata/src/tests/builtins.go
+++ b/testdata/src/tests/builtins.go
@@ -96,6 +96,12 @@ func testYAML() {
 	yaml.Unmarshal(nil, &m)
 	yaml.NewEncoder(nil).Encode(m)
 	yaml.NewDecoder(nil).Decode(&m)
+
+	var tm TextMarshaler
+	yaml.Marshal(tm)
+	yaml.Unmarshal(nil, &tm)
+	yaml.NewEncoder(nil).Encode(tm)
+	yaml.NewDecoder(nil).Decode(&tm)
 }
 
 func testTOML() {