From db892f6b671b0f3975f78471edcd92e479e1db15 Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Tue, 20 Sep 2022 18:35:27 +0200
Subject: [PATCH] Polishing previous commit

---
 .../cloud/function/context/config/FunctionContextUtils.java   | 4 ++++
 1 file changed, 4 insertions(+)

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;
 	}