add selectedCellType to hook

quadratichq · Jun 20, 2023 · ae056a9 · ae056a9
1 parent 5af393b
commit ae056a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ui/menus/CodeEditor/CodeEditor.tsx b/src/ui/menus/CodeEditor/CodeEditor.tsx
@@ -126,7 +126,7 @@ export const CodeEditor = (props: CodeEditorProps) => {
   }, [selectedCell]);
 
   useAlertOnUnsavedChanges(hasUnsavedChanges);
-  useEditorCellHighlights(isValidRef, editorRef, monacoRef);
+  useEditorCellHighlights(isValidRef, editorRef, monacoRef, selectedCell?.type);
   useEditorOnSelectionChange(isValidRef, editorRef);
 
   const closeEditor = ({ skipUnsavedChangesCheck } = { skipUnsavedChangesCheck: false }) => {