diff --git a/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java b/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java
index eadf01c..8d4e42e 100755
--- a/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java
+++ b/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java
@@ -22,6 +22,7 @@
 import javax.activation.DataHandler;
 
 import junit.framework.TestCase;
+import org.apache.axis2.addressing.EndpointReference;
 import org.apache.axis2.context.MessageContext;
 
 /**
@@ -37,7 +38,7 @@
         Object result = newMc.getProperty(key);
         assertEquals(result, "propValue");
     }
-    
+
     // Regression test for SYNAPSE-309
     public void testClonePartiallyWithAttachments() throws Exception {
         MessageContext origMc = new MessageContext();
@@ -52,10 +53,10 @@
     public void testClonePartiallyWithFrom() throws Exception {
         String fromValue = "uri://some-from-value";
         MessageContext origMc = new MessageContext();
-        origMc.setFrom(fromValue);
+        origMc.setFrom(new EndpointReference(fromValue));
         MessageContext newMc = MessageHelper.clonePartially(origMc);
-        Object result = newMc.getFrom();
-        assertEquals(fromValue, result);
+        EndpointReference result = newMc.getFrom();
+        assertEquals(fromValue, result.getAddress());
     }
 
 }