diff --git a/src/main/kotlin/org/jetbrains/research/testspark/actions/TestSparkAction.kt b/src/main/kotlin/org/jetbrains/research/testspark/actions/TestSparkAction.kt
index f80fecfc2..6304b258d 100644
--- a/src/main/kotlin/org/jetbrains/research/testspark/actions/TestSparkAction.kt
+++ b/src/main/kotlin/org/jetbrains/research/testspark/actions/TestSparkAction.kt
@@ -74,12 +74,18 @@ class TestSparkAction : AnAction() {
     /**
      * Updates the state of the action based on the provided event.
      *
-     * @param e the AnActionEvent object representing the event
+     * @param e `AnActionEvent` object representing the event
      */
     override fun update(e: AnActionEvent) {
-        val file = e.dataContext.getData(CommonDataKeys.PSI_FILE)!!
+        val file = e.dataContext.getData(CommonDataKeys.PSI_FILE)
+
+        if (file == null) {
+            e.presentation.isEnabledAndVisible = false
+            return
+        }
+
         val psiHelper = PsiHelperProvider.getPsiHelper(file)
-        e.presentation.isEnabledAndVisible = psiHelper != null && psiHelper.availableForGeneration(e)
+        e.presentation.isEnabledAndVisible = (psiHelper != null) && psiHelper.availableForGeneration(e)
     }
 
     /**