diff --git a/hibernate-reactive-core/src/test/java/org/hibernate/reactive/ReactiveStatelessWithBatchTest.java b/hibernate-reactive-core/src/test/java/org/hibernate/reactive/ReactiveStatelessWithBatchTest.java
index a30cc1364..38e386bf2 100644
--- a/hibernate-reactive-core/src/test/java/org/hibernate/reactive/ReactiveStatelessWithBatchTest.java
+++ b/hibernate-reactive-core/src/test/java/org/hibernate/reactive/ReactiveStatelessWithBatchTest.java
@@ -364,7 +364,7 @@ public void testStageBatchingUpdate(VertxTestContext context) {
 				.thenCompose( v -> getSessionFactory().withStatelessTransaction( s -> s
 						.createQuery( "from GuineaPig p order by p.id", GuineaPig.class )
 						.getResultList()
-						.thenApply( pigs -> {
+						.thenCompose( pigs -> {
 							pigs.get( 0 ).setName( "One updated" );
 							pigs.get( 1 ).setName( "Two updated" );
 							return s.update( 10, pigs.toArray() );
@@ -383,7 +383,7 @@ public void testStageBatchingUpdateMultiple(VertxTestContext context) {
 				.thenCompose( v -> getSessionFactory().withStatelessTransaction( s -> s
 						.createQuery( "from GuineaPig p order by p.id", GuineaPig.class )
 						.getResultList()
-						.thenApply( pigs -> {
+						.thenCompose( pigs -> {
 							pigs.get( 0 ).setName( "One updated" );
 							pigs.get( 1 ).setName( "Two updated" );
 							return s.updateMultiple( pigs );
@@ -402,7 +402,7 @@ public void testStageBatchingUpdateNoBatchSizeParameter(VertxTestContext context
 				.thenCompose( v -> getSessionFactory().withStatelessTransaction( s -> s
 						.createQuery( "from GuineaPig p order by p.id", GuineaPig.class )
 						.getResultList()
-						.thenApply( pigs -> {
+						.thenCompose( pigs -> {
 							pigs.get( 0 ).setName( "One updated" );
 							pigs.get( 1 ).setName( "Two updated" );
 							return s.update( pigs.get( 0 ), pigs.get( 1 ) );