diff --git a/spring-rabbit/src/test/java/org/springframework/amqp/rabbit/core/RabbitTemplatePublisherCallbacksIntegrationTests3.java b/spring-rabbit/src/test/java/org/springframework/amqp/rabbit/core/RabbitTemplatePublisherCallbacksIntegrationTests3.java
index e9ef548a..f86a9a2f 100644
--- a/spring-rabbit/src/test/java/org/springframework/amqp/rabbit/core/RabbitTemplatePublisherCallbacksIntegrationTests3.java
+++ b/spring-rabbit/src/test/java/org/springframework/amqp/rabbit/core/RabbitTemplatePublisherCallbacksIntegrationTests3.java
@@ -65,10 +65,10 @@ public class RabbitTemplatePublisherCallbacksIntegrationTests3 {
 		template.setConfirmCallback((cd, a, c) -> {
 			confirmLatch.countDown();
 			if (confirmLatch.getCount() == 1) {
-				template.convertAndSend(QUEUE1, ((MyCD) cd).payload);
+				template.convertAndSend(QUEUE1, cd.getId());
 			}
 		});
-		template.convertAndSend("bad.exchange", "junk", "foo", new MyCD("foo"));
+		template.convertAndSend("bad.exchange", "junk", "foo", new CorrelationData("foo"));
 		assertThat(confirmLatch.await(10, TimeUnit.SECONDS)).isTrue();
 		assertThat(template.receive(QUEUE1, 10_000)).isNotNull();
 	}
@@ -102,11 +102,13 @@ public class RabbitTemplatePublisherCallbacksIntegrationTests3 {
 		channel2.close();
 		conn.close();
 		assertThat(TestUtils.getPropertyValue(cf, "cachedChannelsNonTransactional", List.class).size()).isEqualTo(2);
-		template.convertAndSend("", QUEUE2 + "junk", "foo", new MyCD("foo"));
+		CorrelationData correlationData = new CorrelationData("foo");
+		template.convertAndSend("", QUEUE2 + "junk", "foo", correlationData);
 		assertThat(returnLatch.await(10, TimeUnit.SECONDS)).isTrue();
 		assertThat(confirmLatch.await(10, TimeUnit.SECONDS)).isTrue();
 		assertThat(cacheCount.get()).isEqualTo(1);
 		assertThat(returnCalledFirst.get()).isTrue();
+		assertThat(correlationData.getReturnedMessage()).isNotNull();
 		cf.destroy();
 	}
 
@@ -130,7 +132,7 @@ public class RabbitTemplatePublisherCallbacksIntegrationTests3 {
 		channel2.close();
 		conn.close();
 		assertThat(TestUtils.getPropertyValue(cf, "cachedChannelsNonTransactional", List.class).size()).isEqualTo(2);
-		template.convertAndSend("", QUEUE2, "foo", new MyCD("foo"));
+		template.convertAndSend("", QUEUE2, "foo", new CorrelationData("foo"));
 		assertThat(confirmLatch.await(10, TimeUnit.SECONDS)).isTrue();
 		assertThat(cacheCount.get()).isEqualTo(1);
 		cf.destroy();
@@ -147,22 +149,11 @@ public class RabbitTemplatePublisherCallbacksIntegrationTests3 {
 		template.setConfirmCallback((cd, a, c) -> {
 			confirmLatch.countDown();
 		});
-		template.convertSendAndReceive("", QUEUE3, "foo", new MyCD("foo"));
-		template.convertSendAndReceive("", QUEUE3, "foo", new MyCD("foo")); // listener not registered
+		template.convertSendAndReceive("", QUEUE3, "foo", new CorrelationData("foo"));
+		template.convertSendAndReceive("", QUEUE3, "foo", new CorrelationData("foo")); // listener not registered
 		assertThat(confirmLatch.await(10, TimeUnit.SECONDS)).isTrue();
 		assertThat(template.receive(QUEUE3, 10_000)).isNotNull();
 		assertThat(template.receive(QUEUE3, 10_000)).isNotNull();
 	}
 
-
-	private static class MyCD extends CorrelationData {
-
-		final String payload;
-
-		MyCD(String payload) {
-			this.payload = payload;
-		}
-
-	}
-
 }