diff --git a/spring-ai-core/src/test/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilderTests.java b/spring-ai-core/src/test/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilderTests.java
index e7de18767..6a4147708 100644
--- a/spring-ai-core/src/test/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilderTests.java
+++ b/spring-ai-core/src/test/java/org/springframework/ai/model/function/DefaultFunctionCallbackBuilderTests.java
@@ -233,7 +233,7 @@ class DefaultFunctionCallbackBuilderTests {
 
 	@Test
 	void whenMethodArgumentTypesIsNullThenThrow() {
-		assertThatThrownBy(() -> FunctionCallback.builder().method("methodName", null))
+		assertThatThrownBy(() -> FunctionCallback.builder().method("methodName", (Class<?>[]) null))
 			.isInstanceOf(IllegalArgumentException.class)
 			.hasMessage("Argument types must not be null");
 	}
diff --git a/spring-ai-core/src/test/java/org/springframework/ai/util/PromptAssertTests.java b/spring-ai-core/src/test/java/org/springframework/ai/util/PromptAssertTests.java
index 54b4b3bd1..28a50cf29 100644
--- a/spring-ai-core/src/test/java/org/springframework/ai/util/PromptAssertTests.java
+++ b/spring-ai-core/src/test/java/org/springframework/ai/util/PromptAssertTests.java
@@ -52,7 +52,8 @@ class PromptAssertTests {
 
 	@Test
 	void whenPlaceholdersIsNullThenThrow() {
-		assertThatThrownBy(() -> PromptAssert.templateHasRequiredPlaceholders(new PromptTemplate("{query}"), null))
+		assertThatThrownBy(
+				() -> PromptAssert.templateHasRequiredPlaceholders(new PromptTemplate("{query}"), (String[]) null))
 			.isInstanceOf(IllegalArgumentException.class)
 			.hasMessageContaining("placeholders cannot be null or empty");
 	}