diff --git a/python/educational-core/testSrc/com/jetbrains/edu/coursecreator/actions/CCTaskFileActionTest.java b/python/educational-core/testSrc/com/jetbrains/edu/coursecreator/actions/CCTaskFileActionTest.java
index c8c12a9d075d..ab862d67d320 100644
--- a/python/educational-core/testSrc/com/jetbrains/edu/coursecreator/actions/CCTaskFileActionTest.java
+++ b/python/educational-core/testSrc/com/jetbrains/edu/coursecreator/actions/CCTaskFileActionTest.java
@@ -45,7 +45,7 @@ public class CCTaskFileActionTest extends CCTestCase {
   @NotNull
   private TestActionEvent getActionEvent(VirtualFile virtualFile, AnAction action) {
     MapDataContext context = new MapDataContext();
-    context.put(CommonDataKeys.VIRTUAL_FILE, virtualFile);
+    context.put(CommonDataKeys.VIRTUAL_FILE_ARRAY, new VirtualFile[]{virtualFile});
     context.put(CommonDataKeys.PROJECT, getProject());
     return new TestActionEvent(context, action);
   }