diff --git a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/FunctionContextUtils.java b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/FunctionContextUtils.java
index 252eebb62..83ec0d4b5 100644
--- a/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/FunctionContextUtils.java
+++ b/spring-cloud-function-context/src/main/java/org/springframework/cloud/function/context/config/FunctionContextUtils.java
@@ -104,6 +104,10 @@ public abstract class FunctionContextUtils {
 				}
 			}
 		}
+
+		if (!(param instanceof ParameterizedType) && definition.hasBeanClass()) {
+			return FunctionTypeUtils.discoverFunctionTypeFromClass(definition.getBeanClass());
+		}
 		return param;
 	}