diff --git a/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java b/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java
index e1b95650769b05b3c1eac155e4cbfa107ede7bb7..01b06c22643f06763fe17179fe114156046dd01d 100644
--- a/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java
+++ b/src/main/java/eu/siacs/conversations/crypto/axolotl/XmppAxolotlMessage.java
@@ -75,6 +75,11 @@ public class XmppAxolotlMessage {
 		public String getPlaintext() {
 			return plaintext;
 		}
+
+		public AxolotlService.XmppAxolotlSession getSession() {
+			return session;
+		}
+
 	}
 
 	public XmppAxolotlMessage(Contact contact, Element axolotlMessage) {
diff --git a/src/main/java/eu/siacs/conversations/entities/Message.java b/src/main/java/eu/siacs/conversations/entities/Message.java
index 336af972153bb1859d5fac4280b42c5cc462593b..b1dffc820687e8893c51146bd0d9531328c5d6b2 100644
--- a/src/main/java/eu/siacs/conversations/entities/Message.java
+++ b/src/main/java/eu/siacs/conversations/entities/Message.java
@@ -670,4 +670,8 @@ public class Message extends AbstractEntity {
 	public boolean isTrusted() {
 		return this.axolotlSession != null && this.axolotlSession.isTrusted();
 	}
+
+	public void setAxolotlSession(AxolotlService.XmppAxolotlSession session) {
+		this.axolotlSession = session;
+	}
 }