From b07f62bfe18c0b27cfaf3e69bbe0beada399b02d Mon Sep 17 00:00:00 2001
From: Oleg Zhurakousky <ozhurakousky@vmware.com>
Date: Thu, 3 Nov 2011 02:10:10 -0400
Subject: [PATCH] INT-2212 polished code removed dependency on removed Message
 when disassociating it from MessageGroupMetadata

---
 .../integration/store/AbstractKeyValueMessageStore.java    | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/spring-integration-core/src/main/java/org/springframework/integration/store/AbstractKeyValueMessageStore.java b/spring-integration-core/src/main/java/org/springframework/integration/store/AbstractKeyValueMessageStore.java
index 493a58377e..3f780204cd 100644
--- a/spring-integration-core/src/main/java/org/springframework/integration/store/AbstractKeyValueMessageStore.java
+++ b/spring-integration-core/src/main/java/org/springframework/integration/store/AbstractKeyValueMessageStore.java
@@ -186,12 +186,9 @@ public abstract class AbstractKeyValueMessageStore extends AbstractMessageGroupS
 			
 			UUID firstId = messageGroupMetadata.firstId();
 			if (firstId != null){
-				Message<?> message = this.removeMessage(firstId);
-				
-				messageGroupMetadata.remove(message.getHeaders().getId());
+				messageGroupMetadata.remove(firstId);
 				this.doStore(MESSAGE_GROUP_KEY_PREFIX + groupId, messageGroupMetadata);
-				
-				return message;
+				return this.removeMessage(firstId);
 			}
 		}
 		return null;