diff --git a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java
index 39cea3a0e..cbde568cb 100644
--- a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java
+++ b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/catalog/SimpleFunctionRegistry.java
@@ -808,7 +808,7 @@ public class SimpleFunctionRegistry implements FunctionRegistry {
 				if (functionInvocationHelper != null && CloudEventMessageUtils.isCloudEvent(((Message) input))) {
 					result = functionInvocationHelper.postProcessResult(result, (Message) input);
 				}
-				if (!FunctionTypeUtils.isCollectionOfMessage(this.outputType)) {
+				if (!(result instanceof Message) && !FunctionTypeUtils.isCollectionOfMessage(this.outputType)) {
 					result = MessageBuilder.withPayload(result).copyHeaders(this.sanitizeHeaders(((Message) input).getHeaders())).build();
 				}
 			}