diff --git a/platform/lang-impl/src/com/intellij/ide/actions/AssociateFileTypeAction.java b/platform/lang-impl/src/com/intellij/ide/actions/AssociateFileTypeAction.java
index fba90ef8f674..2015b86061a3 100644
--- a/platform/lang-impl/src/com/intellij/ide/actions/AssociateFileTypeAction.java
+++ b/platform/lang-impl/src/com/intellij/ide/actions/AssociateFileTypeAction.java
@@ -34,9 +34,8 @@ public class AssociateFileTypeAction extends AnAction {
   @Override
   public void update(AnActionEvent e) {
     Presentation presentation = e.getPresentation();
-    DataContext dataContext = e.getDataContext();
-    VirtualFile file = CommonDataKeys.VIRTUAL_FILE.getData(dataContext);
-    Project project = CommonDataKeys.PROJECT.getData(dataContext);
+    VirtualFile file = e.getData(CommonDataKeys.VIRTUAL_FILE);
+    Project project = e.getProject();
     boolean haveSmthToDo;
     if (project == null || file == null || file.isDirectory()) {
       haveSmthToDo = false;