1 files changed, 18 insertions, 2 deletions
diff --git a/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Task.java b/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Task.java
index e6c085b5d6a1..886add86ceb4 100644
--- a/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Task.java
+++ b/python/edu/course-creator/src/org/jetbrains/plugins/coursecreator/format/Task.java
@@ -2,11 +2,12 @@ package org.jetbrains.plugins.coursecreator.format;
 
 import com.google.gson.annotations.Expose;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.plugins.coursecreator.CCProjectService;
 
 import java.util.HashMap;
 import java.util.Map;
 
-public class Task {
+public class Task implements Comparable{
   @Expose public String name;
   @Expose public Map<String, TaskFile> task_files = new HashMap<String, TaskFile>();
   public int myIndex;
@@ -28,7 +29,8 @@ public class Task {
   }
 
   public TaskFile getTaskFile(@NotNull final String name) {
-    return task_files.get(name);
+    String fileName = CCProjectService.getRealTaskFileName(name);
+    return fileName != null ? task_files.get(fileName) : null;
   }
 
   public void setIndex(int index) {
@@ -38,4 +40,18 @@ public class Task {
   public Map<String, TaskFile> getTaskFiles() {
     return task_files;
   }
+
+  public boolean isTaskFile(String name) {
+    return task_files.get(name) != null;
+  }
+
+  public void setName(String name) {
+    this.name = name;
+  }
+
+  @Override
+  public int compareTo(@NotNull Object o) {
+    Task task = (Task) o;
+    return myIndex - task.getIndex();
+  }
 }