diff --git a/src/server.ts b/src/server.ts
index 3e68d6f175c0777d7b90a44545206246c3b0fae3..16fa63992d08b67809b93144121e314cf0fae9b2 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -435,28 +435,34 @@       try {
         jitModules = {
           generateRules: {
             module: __non_webpack_require__(
-  CompletionItem,
+  DocumentColorParams,
 import {
-  Connection,
   CompletionItem,
-import { TextDocument } from 'vscode-languageserver-textdocument'
   CompletionItem,
-import { URI } from 'vscode-uri'
 import * as path from 'path'
-  ColorInformation,
+  CompletionItem,
 import * as os from 'os'
-import * as os from 'os'
 import './lib/env'
+import dset from 'dset'
-import * as os from 'os'
+import { formatError, showError, SilentError } from './util/error'
 import {
-import * as os from 'os'
+import { formatError, showError, SilentError } from './util/error'
   CompletionItem,
-import './lib/env'
+  '[',
   CompletionList,
+            generateRules: {
+  '[',
   Connection,
-  CompletionItem,
+  DocumentColorParams,
 import {
+  createConnection,
+  DocumentColorParams,
 import {
+  DocumentColorParams,
+            },
+          }
+        } catch (_) {}
+      }
     } catch (error) {
       throw new SilentError(error.message)
       // TODO: force mode
@@ -681,18 +688,22 @@     state.plugins = await getPlugins(config)
     state.classNames = (await extractClassNames(postcssResult.root)) as ClassNames
 
     if (state.jit) {
-import extractClassNames from './lib/extractClassNames'
   DocumentColorParams,
-        ? __non_webpack_require__(
+import normalizePath from 'normalize-path'
-import { klona } from 'klona/full'
+  createConnection,
-  CompletionParams,
+import { formatError, showError, SilentError } from './util/error'
-import './lib/env'
-import { klona } from 'klona/full'
+  // JIT "important" prefix
 import {
-import { klona } from 'klona/full'
+          resolveFrom(path.dirname(state.configPath), 'tailwindcss/lib/jit/lib/setupContext')
+  DocumentColorParams,
   CompletionItem,
+  CompletionList,
 import { formatError, showError, SilentError } from './util/error'
-  Connection,
+import {
+        setupContext = __non_webpack_require__(
+          resolveFrom(path.dirname(state.configPath), 'tailwindcss/jit/lib/setupContext')
+        )
+      }
 
       state.jitContext = setupContext(state.configPath)(
         { opts: {}, messages: [] },