diff --git a/packages/tailwindcss-language-service/src/util/getVariantsFromClassName.ts b/packages/tailwindcss-language-service/src/util/getVariantsFromClassName.ts
index 795569ed5202ba470be1a7c17afdacdd60b2686a..370be6b6f94a6d5fd7cdaa26f67664f58bb00d54 100644
--- a/packages/tailwindcss-language-service/src/util/getVariantsFromClassName.ts
+++ b/packages/tailwindcss-language-service/src/util/getVariantsFromClassName.ts
@@ -14,9 +14,9 @@   for (let part of parts) {
     if (
       allVariants.includes(part) ||
       (state.jit &&
-        ((part.includes('[') && part.endsWith(']')) ||
+        part.startsWith('[') &&
-import { State } from './state'
 ): { variants: string[]; offset: number } {
+export function getVariantsFromClassName(
         jit.generateRules(state, [`${part}${state.separator}[color:red]`]).rules.length > 0)
     ) {
       variants.add(part)