diff --git a/modules/swagger-core/src/test/java/io/swagger/v3/core/resolving/v31/ModelResolverOAS31Test.java b/modules/swagger-core/src/test/java/io/swagger/v3/core/resolving/v31/ModelResolverOAS31Test.java
index 187c1bbe1a..0497e37a22 100644
--- a/modules/swagger-core/src/test/java/io/swagger/v3/core/resolving/v31/ModelResolverOAS31Test.java
+++ b/modules/swagger-core/src/test/java/io/swagger/v3/core/resolving/v31/ModelResolverOAS31Test.java
@@ -241,7 +241,7 @@ public void testFieldArraySchemaAnnotation() {
     @Test(description = "@Pattern correctly handled in type parameters of properties using collections when using oas 3.1.0")
     public void testModelUsingCollectionTypePropertyDoesNotHandlePatternAnnotationForOas31() {
 
-        String expectedYaml = "DtoUsingPatternOnCollection:\n" +
+        String expectedYaml = "ClassWithUsingPatternOnCollection:\n" +
                 "  type: object\n" +
                 "  properties:\n" +
                 "    myField:\n" +
@@ -269,7 +269,7 @@ public void setMyField(List<String> myField) {
     @Test(description = "@Size correctly handled in properties using collections when using oas 3.1.0")
     public void testModelUsingCollectionTypePropertyDoesNotHandleSizeAnnotationForOas31() {
 
-        String expectedYaml = "DtoUsingSizeOnCollection:\n" +
+        String expectedYaml = "ClassWithUsingSizeOnCollection:\n" +
                 "  type: object\n" +
                 "  properties:\n" +
                 "    myField:\n" +