diff --git a/docs/src/main/asciidoc/_project-features-messaging.adoc b/docs/src/main/asciidoc/_project-features-messaging.adoc
index d8f40b9915..fb44412aba 100644
--- a/docs/src/main/asciidoc/_project-features-messaging.adoc
+++ b/docs/src/main/asciidoc/_project-features-messaging.adoc
@@ -526,7 +526,7 @@ include::{tests_path}/spring-cloud-contract-stub-runner-stream/src/test/groovy/o
 Now consider the following Spring Cloud Stream function configuration:
 
 ====
-[source,yaml]
+[source,groovy]
 ----
 include::{tests_path}/spring-cloud-contract-stub-runner-stream/src/test/groovy/org/springframework/cloud/contract/stubrunner/messaging/stream/StreamStubRunnerSpec.groovy[tags=setup,indent=0]
 ----
diff --git a/tests/spring-cloud-contract-stub-runner-stream/src/test/groovy/org/springframework/cloud/contract/stubrunner/messaging/stream/StreamStubRunnerSpec.groovy b/tests/spring-cloud-contract-stub-runner-stream/src/test/groovy/org/springframework/cloud/contract/stubrunner/messaging/stream/StreamStubRunnerSpec.groovy
index 71beec1f73..cca62de6e8 100644
--- a/tests/spring-cloud-contract-stub-runner-stream/src/test/groovy/org/springframework/cloud/contract/stubrunner/messaging/stream/StreamStubRunnerSpec.groovy
+++ b/tests/spring-cloud-contract-stub-runner-stream/src/test/groovy/org/springframework/cloud/contract/stubrunner/messaging/stream/StreamStubRunnerSpec.groovy
@@ -153,7 +153,7 @@ class StreamStubRunnerSpec {
 
 	// Contract from my service that is processing the input message and sending out another message (I'm a producer)
 	Contract myDsl =
-			// tag::sample_dsl[]
+			// tag::sample_producer_dsl[]
 			Contract.make {
 				label 'return_book_2'
 				input { triggeredBy('gotAMessageFromFunction()') }
@@ -163,6 +163,7 @@ class StreamStubRunnerSpec {
 					headers { header('BOOK-NAME', 'foo') }
 				}
 			}
+	// end::sample_producer_dsl[]
 
 	// tag::setup[]
 	@ImportAutoConfiguration(TestChannelBinderConfiguration.class)