diff --git a/spring-integration-cassandra/src/test/java/org/springframework/integration/cassandra/config/CassandraOutboundAdapterParserTests-context.xml b/spring-integration-cassandra/src/test/java/org/springframework/integration/cassandra/config/CassandraOutboundAdapterParserTests-context.xml
index 804fe766eb..80dbbe5558 100644
--- a/spring-integration-cassandra/src/test/java/org/springframework/integration/cassandra/config/CassandraOutboundAdapterParserTests-context.xml
+++ b/spring-integration-cassandra/src/test/java/org/springframework/integration/cassandra/config/CassandraOutboundAdapterParserTests-context.xml
@@ -19,11 +19,12 @@
 	</int:channel>
 
 	<bean id="cassandraTemplate" class="org.mockito.Mockito" factory-method="mock">
-		<constructor-arg value="org.springframework.data.cassandra.core.ReactiveCassandraOperations"/>
+		<constructor-arg value="org.springframework.data.cassandra.core.ReactiveCassandraOperations"
+						 type="java.lang.Class"/>
 	</bean>
 
 	<bean id="writeOptions" class="org.mockito.Mockito" factory-method="mock">
-		<constructor-arg value="org.springframework.data.cassandra.core.InsertOptions"/>
+		<constructor-arg value="org.springframework.data.cassandra.core.InsertOptions" type="java.lang.Class"/>
 	</bean>
 
 	<int-cassandra:outbound-channel-adapter id="outbound1"