diff --git a/gradlew b/gradlew
old mode 100644
new mode 100755
diff --git a/src/main/kotlin/com/chainstaysoftware/testing/ModuleAction.kt b/src/main/kotlin/com/chainstaysoftware/testing/ModuleAction.kt
index c8905d7..5a97b06 100644
--- a/src/main/kotlin/com/chainstaysoftware/testing/ModuleAction.kt
+++ b/src/main/kotlin/com/chainstaysoftware/testing/ModuleAction.kt
@@ -1,5 +1,6 @@
 package com.chainstaysoftware.testing
 
+import com.intellij.openapi.actionSystem.ActionUpdateThread
 import com.intellij.ide.highlighter.JavaFileType
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
@@ -14,7 +15,6 @@ import com.intellij.openapi.roots.ProjectFileIndex
 import com.intellij.psi.PsiFile
 import com.intellij.psi.search.FileTypeIndex
 import com.intellij.psi.search.GlobalSearchScope
-import com.intellij.util.indexing.FileBasedIndex
 
 
 /**
@@ -78,4 +78,8 @@ class ModuleAction : AnAction() {
          .filter { virtualFile -> instance.isInTestSourceContent(virtualFile) }
          .size
    }
+
+   override fun getActionUpdateThread(): ActionUpdateThread {
+       return ActionUpdateThread.EDT
+   }
 }