From 5e3f9fc8ff607fc4e42bd1b02ef0831d3e57c06d Mon Sep 17 00:00:00 2001
From: "Liana.Bakradze" <liana.bakradze@jetbrains.com>
Date: Fri, 11 Nov 2016 14:40:59 +0300
Subject: [PATCH] fix add subtask from test files

---
 .../edu/coursecreator/actions/CCNewSubtaskAction.java       | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/python/educational-core/course-creator/src/com/jetbrains/edu/coursecreator/actions/CCNewSubtaskAction.java b/python/educational-core/course-creator/src/com/jetbrains/edu/coursecreator/actions/CCNewSubtaskAction.java
index 663ac14cbd82..ae3558f524ed 100644
--- a/python/educational-core/course-creator/src/com/jetbrains/edu/coursecreator/actions/CCNewSubtaskAction.java
+++ b/python/educational-core/course-creator/src/com/jetbrains/edu/coursecreator/actions/CCNewSubtaskAction.java
@@ -25,7 +25,6 @@ import com.jetbrains.edu.learning.StudyUtils;
 import com.jetbrains.edu.learning.core.EduNames;
 import com.jetbrains.edu.learning.courseFormat.Course;
 import com.jetbrains.edu.learning.courseFormat.Task;
-import com.jetbrains.edu.learning.courseFormat.TaskFile;
 import org.jetbrains.annotations.NotNull;
 
 
@@ -50,11 +49,10 @@ public class CCNewSubtaskAction extends DumbAwareAction {
   }
 
   public static void addSubtask(@NotNull VirtualFile virtualFile, @NotNull Project project) {
-    TaskFile taskFile = StudyUtils.getTaskFile(project, virtualFile);
-    if (taskFile == null) {
+    Task task = StudyUtils.getTaskForFile(project, virtualFile);
+    if (task == null) {
       return;
     }
-    Task task = taskFile.getTask();
     VirtualFile taskDir = task.getTaskDir(project);
     if (taskDir == null) {
       return;