diff --git a/src/test/java/org/springframework/data/r2dbc/core/DefaultDatabaseClientUnitTests.java b/src/test/java/org/springframework/data/r2dbc/core/DefaultDatabaseClientUnitTests.java
index 21c39fc6..5c88fec9 100644
--- a/src/test/java/org/springframework/data/r2dbc/core/DefaultDatabaseClientUnitTests.java
+++ b/src/test/java/org/springframework/data/r2dbc/core/DefaultDatabaseClientUnitTests.java
@@ -333,7 +333,7 @@ public class DefaultDatabaseClientUnitTests {
 				.connectionFactory(connectionFactory)
 				.dataAccessStrategy(new DefaultReactiveDataAccessStrategy(PostgresDialect.INSTANCE)).build();
 
-		when(result.getRowsUpdated()).thenReturn(Mono.empty());
+		when(result.getRowsUpdated()).thenReturn(Mono.empty(), Mono.just(2), Flux.just(1, 2, 3));
 
 		databaseClient.execute("DROP TABLE tab;") //
 				.fetch() //
@@ -342,8 +342,6 @@ public class DefaultDatabaseClientUnitTests {
 				.expectNextCount(1) //
 				.verifyComplete();
 
-		when(result.getRowsUpdated()).thenReturn(Mono.just(2));
-
 		databaseClient.execute("DROP TABLE tab;") //
 				.fetch() //
 				.rowsUpdated() //
@@ -351,8 +349,6 @@ public class DefaultDatabaseClientUnitTests {
 				.expectNextCount(1) //
 				.verifyComplete();
 
-		when(result.getRowsUpdated()).thenReturn(Flux.just(1, 2, 3));
-
 		databaseClient.execute("DROP TABLE tab;") //
 				.fetch() //
 				.rowsUpdated() //