diff --git a/media/media_test.go b/media/media_test.go
index 8b37319..0a310f2 100644
--- a/media/media_test.go
+++ b/media/media_test.go
@@ -173,7 +173,7 @@ func TestNormalizeOverCount(t *testing.T) {
 			case llm.ImagePart:
 				imgs = append(imgs, v)
 			case llm.TextPart:
-				if strings.Contains(v.Text, "omitted") {
+				if v.Text == imageOverflowPlaceholder {
 					placeholders++
 				}
 			}