diff --git a/spring-cloud-function-samples/function-sample-aws-custom/src/test/java/com/example/ContainerTests.java b/spring-cloud-function-samples/function-sample-aws-custom/src/test/java/com/example/ContainerTests.java
index 410b92559..764c39166 100644
--- a/spring-cloud-function-samples/function-sample-aws-custom/src/test/java/com/example/ContainerTests.java
+++ b/spring-cloud-function-samples/function-sample-aws-custom/src/test/java/com/example/ContainerTests.java
@@ -52,7 +52,7 @@ public class ContainerTests {
 						String.class);
 				return result;
 			}, result -> result != null);
-			assertThat(response.getBody()).contains("hi foo!");
+			assertThat(response.getBody()).isEqualTo("\"FOO\"");
 			assertThat(response.getHeaders()).containsKey("X-Amzn-Requestid");
 		}
 		String output = consumer.toUtf8String();