1 files changed, 3 insertions, 2 deletions
diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/AgentMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/AgentMessage.java
index ba15a7a..b71831f 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/AgentMessage.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/AgentMessage.java
@@ -1,7 +1,6 @@
 package pl.edu.mimuw.cloudatlas.agent.messages;
 
-public class AgentMessage {
-    public AgentModule correctType;
+public abstract class AgentMessage {
 
     public enum AgentModule {
         TIMER_SCHEDULER,
@@ -30,6 +29,8 @@ public class AgentMessage {
         this.timestamp = System.currentTimeMillis() / 1000L;
     }
 
+    public abstract AgentModule getCorrectMessageType();
+
     public String getMessageId() {
         return messageId;
     }