diff --git a/integrationtest/mapper/orm/src/test/java/org/hibernate/search/integrationtest/mapper/orm/automaticindexing/association/bytype/manytomany/ownedbycontaining/AutomaticIndexingManyToManyOwnedByContainingMapKeysBaseIT.java b/integrationtest/mapper/orm/src/test/java/org/hibernate/search/integrationtest/mapper/orm/automaticindexing/association/bytype/manytomany/ownedbycontaining/AutomaticIndexingManyToManyOwnedByContainingMapKeysBaseIT.java
index 999c18d9c24..be5073713c1 100644
--- a/integrationtest/mapper/orm/src/test/java/org/hibernate/search/integrationtest/mapper/orm/automaticindexing/association/bytype/manytomany/ownedbycontaining/AutomaticIndexingManyToManyOwnedByContainingMapKeysBaseIT.java
+++ b/integrationtest/mapper/orm/src/test/java/org/hibernate/search/integrationtest/mapper/orm/automaticindexing/association/bytype/manytomany/ownedbycontaining/AutomaticIndexingManyToManyOwnedByContainingMapKeysBaseIT.java
@@ -214,7 +214,7 @@ public static class ContainingEntity {
 				extraction = @ContainerExtraction(BuiltinContainerExtractors.MAP_KEY),
 				targetType = ContainedEntity.class
 		)
-		private Map<Object, String> containedIndexedEmbeddedWithCast = new LinkedHashMap<>();
+		private Map<ContainedEntity, String> containedIndexedEmbeddedWithCast = new LinkedHashMap<>();
 
 		@IndexedEmbedded
 		@Embedded
@@ -290,7 +290,7 @@ public Map<ContainedEntity, String> getContainedUsedInCrossEntityDerivedProperty
 			return containedUsedInCrossEntityDerivedProperty;
 		}
 
-		public Map<Object, String> getContainedIndexedEmbeddedWithCast() {
+		public Map<ContainedEntity, String> getContainedIndexedEmbeddedWithCast() {
 			return containedIndexedEmbeddedWithCast;
 		}