diff --git a/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/JpaDefaultTest.java b/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/JpaDefaultTest.java
index 0908c5a8b7..d3705519e2 100644
--- a/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/JpaDefaultTest.java
+++ b/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/JpaDefaultTest.java
@@ -136,6 +136,7 @@ private void verifyProject() throws Exception {
 		String generatedPersonJavaFileContents = new String(
 				Files.readAllBytes(generatedPersonJavaFile.toPath()));
 		assertTrue(generatedPersonJavaFileContents.contains("import jakarta.persistence.Entity;"));
+		assertTrue(generatedPersonJavaFileContents.contains("public class Person "));
 	}
 	
 	private void addHibernateToolsPluginLine(StringBuffer gradleBuildFileContents) {
diff --git a/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/NoAnnotationsTest.java b/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/NoAnnotationsTest.java
index 4b8b0e634f..3176e86039 100644
--- a/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/NoAnnotationsTest.java
+++ b/gradle/plugin/src/functionalTest/java/org/hibernate/tool/gradle/java/NoAnnotationsTest.java
@@ -137,6 +137,7 @@ private void verifyProject() throws Exception {
 		String generatedPersonJavaFileContents = new String(
 				Files.readAllBytes(generatedPersonJavaFile.toPath()));
 		assertFalse(generatedPersonJavaFileContents.contains("import jakarta.persistence.Entity;"));
+		assertTrue(generatedPersonJavaFileContents.contains("public class Person "));
 	}
 	
 	private void addHibernateToolsPluginLine(StringBuffer gradleBuildFileContents) {