diff --git a/packages/tailwindcss-language-service/src/diagnostics/getRecommendedVariantOrderDiagnostics.ts b/packages/tailwindcss-language-service/src/diagnostics/getRecommendedVariantOrderDiagnostics.ts
index 72a71efa4b5a0bc8e6a827c311e1f641ac028f3b..82385d979baf577950c361e46cea6d97741e2994 100644
--- a/packages/tailwindcss-language-service/src/diagnostics/getRecommendedVariantOrderDiagnostics.ts
+++ b/packages/tailwindcss-language-service/src/diagnostics/getRecommendedVariantOrderDiagnostics.ts
@@ -30,9 +30,10 @@       if (rules.length === 0) {
         return
       }
 
-      let order = state.jitContext.variantOrder ?? state.jitContext.offsets.variantOffsets
       let { variants, offset } = getVariantsFromClassName(state, className.className)
-      let sortedVariants = [...variants].sort((a, b) => jit.bigSign(order.get(b) - order.get(a)))
+      let sortedVariants = [...variants].sort((a, b) =>
+        jit.bigSign(state.jitContext.variantOrder.get(b) - state.jitContext.variantOrder.get(a))
+      )
 
       if (!equalExact(variants, sortedVariants)) {
         diagnostics.push({