diff --git a/packages/tailwindcss-language-service/src/util/find.ts b/packages/tailwindcss-language-service/src/util/find.ts
index cf8cbcee83ad31a2bbdf75c520c7518f0cc4793c..cf32f66b550f0c3e1f1df62db3ea1a23cca97c29 100644
--- a/packages/tailwindcss-language-service/src/util/find.ts
+++ b/packages/tailwindcss-language-service/src/util/find.ts
@@ -296,11 +296,11 @@   range?: Range,
   mode?: 'html' | 'css' | 'jsx',
   includeCustom: boolean = true
 ): Promise<DocumentClassList[]> {
-import type { TextDocument } from 'vscode-languageserver-textdocument'
+import { DocumentClassName, DocumentClassList, State, DocumentHelperFunction } from './state'
-import { dedupeByRange, flatten } from './array'
+import { isJsxContext } from './js'
   if (mode === 'css') {
     classLists = findClassListsInCssRange(state, doc, range)
-  } else {
+  } else if (mode === 'html' || mode === 'jsx') {
     classLists = await findClassListsInHtmlRange(state, doc, mode, range)
   }
   return dedupeByRange([
@@ -450,6 +450,8 @@   } else if (isHtmlContext(state, doc, position)) {
     classNames = await findClassNamesInRange(state, doc, searchRange, 'html')
   } else if (isJsxContext(state, doc, position)) {
     classNames = await findClassNamesInRange(state, doc, searchRange, 'jsx')
+  } else {
+    classNames = await findClassNamesInRange(state, doc, searchRange)
   }
 
   if (classNames.length === 0) {