diff --git a/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ChatToolWindowTabPanel.java b/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ChatToolWindowTabPanel.java
index 9e7f7ce2..7e528169 100644
--- a/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ChatToolWindowTabPanel.java
+++ b/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/ChatToolWindowTabPanel.java
@@ -308,7 +308,8 @@ public class ChatToolWindowTabPanel implements Disposable {
       var selectionModel = editor.getSelectionModel();
       var selectedText = selectionModel.getSelectedText();
       if (selectedText != null && !selectedText.isEmpty()) {
-        var fileExtension = FileUtil.getFileExtension(editor.getVirtualFile().getName());
+        var fileExtension = FileUtil.getFileExtension(
+            ((EditorEx) editor).getVirtualFile().getName());
         highlightedTextMd = format("\n```%s\n%s\n```\n", fileExtension, selectedText);
         selectionModel.removeSelection();
       }
diff --git a/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/actionprocessor/CodeActionProcessor.java b/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/actionprocessor/CodeActionProcessor.java
index 66980439..e5a6b7bb 100644
--- a/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/actionprocessor/CodeActionProcessor.java
+++ b/src/main/java/ee/carlrobert/codegpt/toolwindow/chat/actionprocessor/CodeActionProcessor.java
@@ -1,6 +1,7 @@
 package ee.carlrobert.codegpt.toolwindow.chat.actionprocessor;
 
 import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.editor.ex.EditorEx;
 import ee.carlrobert.codegpt.conversations.message.Message;
 import ee.carlrobert.codegpt.ui.textarea.AppliedActionInlay;
 import ee.carlrobert.codegpt.ui.textarea.AppliedCodeActionInlay;
@@ -20,7 +21,8 @@ public class CodeActionProcessor implements ActionProcessor {
   private void processCodeAction(AppliedCodeActionInlay action, Editor editor,
       StringBuilder promptBuilder) {
     promptBuilder
-        .append("\n```%s\n".formatted(FileUtil.getFileExtension(editor.getVirtualFile().getName())))
+        .append("\n```%s\n".formatted(
+            FileUtil.getFileExtension(((EditorEx) editor).getVirtualFile().getName())))
         .append(action.getCode())
         .append("\n```\n");
   }