diff --git a/src/main/java/ee/carlrobert/codegpt/actions/DisableCompletionsAction.java b/src/main/java/ee/carlrobert/codegpt/actions/DisableCompletionsAction.java
index 4aa7382b..13e444d6 100644
--- a/src/main/java/ee/carlrobert/codegpt/actions/DisableCompletionsAction.java
+++ b/src/main/java/ee/carlrobert/codegpt/actions/DisableCompletionsAction.java
@@ -1,8 +1,8 @@
 package ee.carlrobert.codegpt.actions;
 
+import com.intellij.openapi.actionSystem.AnAction;
 import com.intellij.openapi.actionSystem.AnActionEvent;
 import com.intellij.openapi.application.ApplicationManager;
-import com.intellij.openapi.project.DumbAwareAction;
 import ee.carlrobert.codegpt.codecompletions.CodeGPTEditorManager;
 import ee.carlrobert.codegpt.settings.configuration.ConfigurationState;
 import org.jetbrains.annotations.NotNull;
@@ -10,7 +10,7 @@ import org.jetbrains.annotations.NotNull;
 /**
  * Disables code-completion.<br/> Publishes message to {@link CodeCompletionEnabledListener#TOPIC}
  */
-public class DisableCompletionsAction extends DumbAwareAction {
+public class DisableCompletionsAction extends AnAction {
 
   @Override
   public void actionPerformed(@NotNull AnActionEvent e) {
diff --git a/src/main/java/ee/carlrobert/codegpt/actions/EnableCompletionsAction.java b/src/main/java/ee/carlrobert/codegpt/actions/EnableCompletionsAction.java
index e74fe41f..39fe132b 100644
--- a/src/main/java/ee/carlrobert/codegpt/actions/EnableCompletionsAction.java
+++ b/src/main/java/ee/carlrobert/codegpt/actions/EnableCompletionsAction.java
@@ -1,15 +1,15 @@
 package ee.carlrobert.codegpt.actions;
 
+import com.intellij.openapi.actionSystem.AnAction;
 import com.intellij.openapi.actionSystem.AnActionEvent;
 import com.intellij.openapi.application.ApplicationManager;
-import com.intellij.openapi.project.DumbAwareAction;
 import ee.carlrobert.codegpt.settings.configuration.ConfigurationState;
 import org.jetbrains.annotations.NotNull;
 
 /**
  * Enables code-completion.<br/> Publishes message to {@link CodeCompletionEnabledListener#TOPIC}
  */
-public class EnableCompletionsAction extends DumbAwareAction {
+public class EnableCompletionsAction extends AnAction {
 
   @Override
   public void actionPerformed(@NotNull AnActionEvent e) {