1 files changed, 12 insertions, 6 deletions

diff --git a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java
index 9a558b7..333b381 100644
--- a/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java
+++ b/src/main/java/pl/edu/mimuw/cloudatlas/agent/messages/TimerSchedulerMessage.java
@@ -1,14 +1,20 @@
 package pl.edu.mimuw.cloudatlas.agent.messages;
 
-import java.util.TimerTask;
+import pl.edu.mimuw.cloudatlas.agent.modules.TimerScheduledTask;
 
 public class TimerSchedulerMessage extends AgentMessage {
     private String requestId;
     private long delay;
     private long baseTime;
-    private TimerTask task;
-
-    public TimerSchedulerMessage(String messageId, AgentModule destinationModule, long timestamp, String requestId, long delay, long baseTime, TimerTask task) {
+    private TimerScheduledTask task;
+
+    public TimerSchedulerMessage(String messageId,
+                                 AgentModule destinationModule,
+                                 long timestamp,
+                                 String requestId,
+                                 long delay,
+                                 long baseTime,
+                                 TimerScheduledTask task) {
         super(messageId, destinationModule, timestamp);
         this.requestId = requestId;
         this.delay = delay;
@@ -32,11 +38,11 @@ public class TimerSchedulerMessage extends AgentMessage {
         this.baseTime = baseTime;
     }
 
-    public TimerTask getTask() {
+    public TimerScheduledTask getTask() {
         return task;
     }
 
-    public void setTask(TimerTask task) {
+    public void setTask(TimerScheduledTask task) {
         this.task = task;
     }