1 files changed, 6 insertions, 9 deletions
diff --git a/hi-ui/src/store/messages.js b/hi-ui/src/store/messages.js
index cefd725..f9e0856 100644
--- a/hi-ui/src/store/messages.js
+++ b/hi-ui/src/store/messages.js
@@ -7,13 +7,8 @@ export class Messages {
 		return this.channels[channel] || [];
 	}
 
-	addMessage(message) {
-		let {
-			channel,
-			...payload
-		} = message;
-		let channel_id = channel.id;
-		this.updateChannel(channel_id, (messages) => [...messages, payload]);
+	addMessage(channel, at, sender, body) {
+		this.updateChannel(channel, (messages) => [...messages, { at, sender, body }]);
 		return this;
 	}
 
@@ -23,8 +18,10 @@ export class Messages {
 	}
 
 
-	deleteMessage(channel, message) {
-		this.updateChannel(channel, (messages) => messages.filter((msg) => msg.id != message));
+	deleteMessage(message) {
+		for (let channel in this.channels) {
+			this.updateChannel(channel, (messages) => messages.filter((msg) => msg.id != message));
+		}
 		return this;
 	}