diff --git a/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java b/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java
index 14a4cd355..c83ec89d1 100644
--- a/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java
+++ b/core/spring-cloud-stream/src/test/java/org/springframework/cloud/stream/function/FunctionBatchingTests.java
@@ -287,7 +287,7 @@ public class FunctionBatchingTests {
 		public Function<Message<List<Person>>, Person> func() {
 			return x -> {
 				Object o = x.getPayload().get(2);
-				assertThat(o).isInstanceOf(KafkaNull.class);
+				assertThat(o).isNull();
 				return (Person) x.getPayload().get(0);
 			};
 		}