diff --git a/framework/framework/package.json b/framework/framework/package.json
index 12ed76220..b6d2f4e62 100644
--- a/framework/framework/package.json
+++ b/framework/framework/package.json
@@ -1,6 +1,6 @@
 {
     "name": "@hydrooj/framework",
-    "version": "0.0.2",
+    "version": "0.0.3",
     "main": "index.ts",
     "peerDependencies": {
         "cordis": "*"
diff --git a/framework/framework/server.ts b/framework/framework/server.ts
index 09897f36b..86c35710a 100644
--- a/framework/framework/server.ts
+++ b/framework/framework/server.ts
@@ -8,7 +8,7 @@ import Koa from 'koa';
 import Body from 'koa-body';
 import Compress from 'koa-compress';
 import { Shorty } from 'shorty.js';
-import WebSocket from 'ws';
+import { WebSocketServer } from 'ws';
 import {
     Counter, errorMessage, isClass, Logger, parseMemoryMB,
 } from '@hydrooj/utils/lib/utils';
@@ -91,7 +91,7 @@ export const koa = new Koa<Koa.DefaultState, KoaContext>({
 });
 export const router = new Router();
 export const httpServer = http.createServer(koa.callback());
-export const wsServer = new WebSocket.Server({ server: httpServer });
+export const wsServer = new WebSocketServer({ server: httpServer });
 koa.on('error', (error) => {
     if (error.code !== 'EPIPE' && error.code !== 'ECONNRESET' && !error.message.includes('Parse Error')) {
         logger.error('Koa app-level error', { error });