diff --git a/idea/src/main/kotlin/fi/jyu/intellij/comtest/idea/actions/ComTestGenerateJavaFileActionBase.kt b/idea/src/main/kotlin/fi/jyu/intellij/comtest/idea/actions/ComTestGenerateJavaFileActionBase.kt
index d3d61d98ab586b4036bb6a4b35f7a30112ffb4bd..4dd5d2d193c0d3beb2994863d9543f021a403287 100644
--- a/idea/src/main/kotlin/fi/jyu/intellij/comtest/idea/actions/ComTestGenerateJavaFileActionBase.kt
+++ b/idea/src/main/kotlin/fi/jyu/intellij/comtest/idea/actions/ComTestGenerateJavaFileActionBase.kt
@@ -20,7 +20,6 @@ import com.intellij.psi.PsiManager
 import fi.jyu.intellij.comtest.common.util.Notifier
 import fi.jyu.intellij.comtest.common.util.runWithRefresh
 import fi.jyu.intellij.comtest.idea.ComTestRunner
-import org.jetbrains.concurrency.await
 
 /**
  * Action to run ComTest on the currently selected/open Java file.
@@ -46,10 +45,8 @@ abstract class ComTestGenerateJavaFileActionBase : AnAction() {
                 false
             }
         if (!hasJUnit) {
-            suspend {
-                JavaProjectModelModificationService.getInstance(proj)
-                    .addDependency(fileModule, JUnitExternalLibraryDescriptor.JUNIT4).await()
-            }
+            JavaProjectModelModificationService.getInstance(proj)
+                .addDependency(fileModule, JUnitExternalLibraryDescriptor.JUNIT4)
         }
         return true
     }