1 files changed, 7 insertions, 6 deletions
diff --git a/src/main/java/com/juick/service/activities/UpdateEvent.java b/src/main/java/com/juick/service/activities/UpdateEvent.java
index c5cceb24..879d8005 100644
--- a/src/main/java/com/juick/service/activities/UpdateEvent.java
+++ b/src/main/java/com/juick/service/activities/UpdateEvent.java
@@ -17,31 +17,32 @@
 
 package com.juick.service.activities;
 
+import com.juick.model.Message;
 import com.juick.model.User;
 import org.springframework.context.ApplicationEvent;
 
 public class UpdateEvent extends ApplicationEvent {
     private final User actor;
-    private final String messageUri;
+    private final Message message;
     /**
      * Create a new {@code ApplicationEvent}.
      *
      * @param source the object on which the event initially occurred or with
      *               which the event is associated (never {@code null})
      * @param actor the event author
-     * @param messageUri the object's id
+     * @param message the object
      */
-    public UpdateEvent(Object source, User actor, String messageUri) {
+    public UpdateEvent(Object source, User actor, Message message) {
         super(source);
         this.actor = actor;
-        this.messageUri = messageUri;
+        this.message = message;
     }
 
     public User getUser() {
         return actor;
     }
 
-    public String getMessageUri() {
-        return messageUri;
+    public Message getMessage() {
+        return message;
     }
 }